Update app.py

app.py

@@ -22,7 +22,7 @@ model = model.merge_and_unload()
 
 
 dataset = load_dataset("allenai/sciq")
-random_test_samples = dataset["test"].select(random.sample(range(0, len(dataset["test"])), 10))
+random_test_samples = dataset["test"].select(random.sample(range(0, len(dataset["test"])), 1))
 
 examples = []
 for row in random_test_samples:
@@ -67,8 +67,6 @@ answer_gr = gr.Text(
     value="yellow"
 )
 
-button = gr.Button("Generate", elem_id="send-btn", visible=True)
-
 context_output_gr = gr.Text(
     label="Output"
 )
@@ -88,15 +86,6 @@ distractor3_output_gr = gr.Text(
     label="Output"
 )
 
-# button.click(
-#             predict,
-#             [
-#                 support_gr,
-#                 answer_gr,
-#             ],
-#             outputs=[context_output_gr, answer_output_gr, question_output_gr, distractor1_output_gr, distractor2_output_gr, distractor3_output_gr],
-#         )
-
 gr.Interface(
     fn=predict,
     inputs=[support_gr, answer_gr],