Update app.py

app.py

@@ -19,8 +19,19 @@ def get_beam_output(sentence):
                                     num_beams=5,
                                     early_stopping=True
                                     )
-    tokenizer.decode(beam_output[0], skip_special_tokens=True)
+    output=tokenizer.decode(beam_output[0], skip_special_tokens=True)
+    return output
 
 
+txt1 = gr.Textbox(
+            label="Input",
+            lines=3,
+        )
+
+txt2 = gr.Textbox(
+            label="Output",
+            lines=10,
+        )
+
 demo = gr.Interface(fn=get_beam_output, inputs="text", outputs="text")
 demo.launch()