Update app.py

app.py

@@ -5,13 +5,12 @@ from datasets import load_dataset
 from peft import PeftModel
 import os
 
-title = "Ask Rick a Question"
+title = "Gemma-2b SciQ"
 description = """
-The bot was trained to answer questions based on Rick and Morty dialogues. Ask Rick anything!
-<img src="https://huggingface.co/spaces/course-demos/Rick_and_Morty_QA/resolve/main/rick.png" width=200px>
+Gemma-2b fine-tuned on SciQ
 """
 
-article = "Check out [the original Rick and Morty Bot](https://huggingface.co/spaces/kingabzpro/Rick_and_Morty_Bot) that this demo is based off of."
+article = "GitHub repository: <ur>"
 
 model_id = "google/gemma-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ.get("HF_TOKEN"))
@@ -22,7 +21,7 @@ model = model.merge_and_unload()
 
 
 dataset = load_dataset("allenai/sciq")
-random_test_samples = dataset["test"].select(random.sample(range(0, len(dataset["test"])), 1))
+random_test_samples = dataset["test"].select(random.sample(range(0, len(dataset["test"])), 10))
 
 examples = []
 for row in random_test_samples:
@@ -52,38 +51,36 @@ def predict(context, answer):
             split_outputs[5],
         )
         
-    return None
+    return ("ERROR: " + decoded_outputs, None, None, None, None, None)
     
 
 support_gr = gr.TextArea(
     label="Context",
-    info="Make sure you use proper punctuation.",
     value="Bananas are yellow and curved."
 )
 
 answer_gr = gr.Text(
-    label="Answer optional",
-    info="Make sure you use proper punctuation.",
+    label="Answer (optional)",
     value="yellow"
 )
 
 context_output_gr = gr.Text(
-    label="Output"
+    label="Context"
 )
 answer_output_gr = gr.Text(
-    label="Output"
+    label="Answer"
 )
 question_output_gr = gr.Text(
-    label="Output"
+    label="Question"
 )
 distractor1_output_gr = gr.Text(
-    label="Output"
+    label="Distractor 1"
 )
 distractor2_output_gr = gr.Text(
-    label="Output"
+    label="Distractor 2"
 )
 distractor3_output_gr = gr.Text(
-    label="Output"
+    label="Distractor 3"
 )
 
 gr.Interface(