Update app.py

app.py

@@ -59,6 +59,7 @@ def generate_qa(context, recommended_word, temperature, top_p, num_samples=3):
     generated_text = tokenizer.batch_decode(output, skip_special_tokens=True)
     return generated_text
 
+# Create the Gradio interface with sliders for temperature and top-p
 # Create the Gradio interface with sliders for temperature and top-p
 iface = gr.Interface(
     fn=generate_qa,
@@ -66,13 +67,12 @@ iface = gr.Interface(
         "text",
         gr.inputs.Radio(recommended_words, label="Choose a Recommended Word"),
         gr.inputs.Slider(minimum=0.2, maximum=2, default=1, step=0.1, label="Temperature"),
-        gr.inputs.Slider(minimum=0.1, maximum=1, default=0.8, step=0.1, label="Top-p"),
-        gr.inputs.Number(default=3, label="Number of Samples")
+        gr.inputs.Slider(minimum=0.1, maximum=1, default=0.8, step=0.1, label="Top-p")
     ],
-    outputs=gr.outputs.Textbox(label="Generated Sentences", type="list"),
+    outputs="text",
     title="Question Generation and Answering",
-    description="Enter a context, choose a recommended word, adjust temperature and top-p, and set the number of samples. The model will generate questions and answers.",
+    description="Enter a context, choose a recommended word, and adjust temperature and top-p. The model will generate a question and answer.",
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()