Update app.py

app.py

@@ -33,7 +33,7 @@ pipe = pipeline(
 
 @spaces.GPU
 def infer(prompt):
-    return pipe(prompt, max_new_tokens=1, do_sample=True, temperature=1.0, return_full_text=False)[0]["generated_text"]
+    return prompt + pipe(prompt, max_new_tokens=1, do_sample=True, temperature=1.0, return_full_text=False)[0]["generated_text"]
 
 default_experiment = """You will take part in a Social Prediction Game.
 You will observe a Player playing against an Opponent.
@@ -53,7 +53,6 @@ with gr.Blocks(
     fill_width=True,
     css="""
     #prompt-box textarea {height:400px}
-    #answer-box textarea {height:400px}
     #info-box {margin-bottom: 1rem}      /* a little spacing */
     """
 ) as demo:
@@ -79,19 +78,16 @@ with gr.Blocks(
         elem_id="info-box",
     )
 
-    # --------------------------------------
-
-    with gr.Row(equal_height=True):
-        inp = gr.Textbox(
-            label="Prompt", elem_id="prompt-box",
-            lines=20, max_lines=24, scale=3, value=default_experiment
-        )
-        outp = gr.Textbox(
-            label="Response", elem_id="answer-box",
-            lines=1, interactive=False, scale=3
-        )
+    inp = gr.Textbox(
+        label="Prompt",
+        elem_id="prompt-box",
+        lines=20,
+        max_lines=24,
+        scale=3,
+        value=default_experiment,
+    )
 
     run = gr.Button("Run")
-    run.click(infer, inp, outp)
+    run.click(infer, inp, inp)
 
 demo.queue().launch()