feat: Add prompt picker and some ui enhancements

app.py

@@ -11,6 +11,33 @@ model.eval()
 
 LABELS = ["A", "B", "C", "D", "E"]
 
+EXAMPLES = [
+    ["""Prompt: What is the function of mammary glands in mammals?
+A: Mammary glands produce milk to feed the young.
+B: Mammary glands help mammals draw air into the lungs.
+C: Mammary glands help mammals breathe with lungs.
+D: Mammary glands excrete nitrogenous waste as urea.
+E: Mammary glands separate oxygenated and deoxygenated blood in the mammalian heart."""],
+    ["""Prompt: Who was the first to determine the velocity of a star moving away from the Earth using the Doppler effect?
+A: Fraunhofer
+B: William Huggins
+C: Hippolyte Fizeau
+D: Vogel and Scheiner
+E: None of the above"""],
+    ["""Prompt: What is the effect generated by a spinning superconductor?
+A: An electric field, precisely aligned with the spin axis.
+B: A magnetic field, randomly aligned with the spin axis.
+C: A magnetic field, precisely aligned with the spin axis.
+D: A gravitational field, randomly aligned with the spin axis.
+E: A gravitational field, precisely aligned with the spin axis."""],
+    ["""Prompt: What is bollard pull primarily used for measuring?
+A: The weight of heavy machinery
+B: The speed of locomotives
+C: The distance traveled by a truck
+D: The strength of tugboats
+E: The height of a ballast tractor"""],
+]
+
 
 @spaces.GPU
 def predict(text):
@@ -43,10 +70,7 @@ def predict(text):
         probs = torch.softmax(model(**inputs).logits, dim=1)[0]
 
     prediction = LABELS[torch.argmax(probs).item()]
-    confidence = {
-        label: f"{prob.item() * 100:.2f}%"
-        for label, prob in zip(LABELS, probs)
-    }
+    confidence = {label: prob.item() for label, prob in zip(LABELS, probs)}
     return prediction, confidence
 
 
@@ -64,8 +88,16 @@ E: Rome"""
     ),
     outputs=[
         gr.Textbox(label="Predicted Answer"),
-        gr.JSON(label="Confidence Scores"),
+        gr.Label(label="Confidence Scores", num_top_classes=5),
+    ],
+    examples=EXAMPLES,
+    example_labels=[
+        "What is the function of mammary glands in mammals?",
+        "Who was the first to determine the velocity of a star moving away from the Earth using the Doppler effect?",
+        "What is the effect generated by a spinning superconductor?",
+        "What is bollard pull primarily used for measuring?",
     ],
+    cache_examples=False,
     title="MCQ Solver",
     description="""
 A multiple-choice question answering system fine-tuned from Google's ELECTRA Base Discriminator model (`google/electra-base-discriminator`).
@@ -78,8 +110,9 @@ B: <option B>
 C: <option C>
 D: <option D>
 E: <option E>
-"""
+""",
+    flagging_mode="never",
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Soft())