final push for now

app.py

@@ -32,7 +32,8 @@ def model_classify(radio, im):
         outputs = model(**inputs)
         logits = outputs.logits
         predicted_class_idx = logits.argmax(-1).item()
-        return model.config.id2label[predicted_class_idx], predicted_class_idx, True
+        modelclass=model.config.id2label[predicted_class_idx]
+        return  modelclass.split(',')[0], predicted_class_idx, True
     else:
         return None, None, False
 
@@ -61,15 +62,14 @@ def check_score(pred, truth, current_score, total_score, has_guessed):
 
 
 
-def compare_score(userclass, model_class, truth, has_guessed):
-    print(model_class)
-    prediction= classes[int(model_class)]
-    if userclass == classes[int(truth)] == prediction:
-        return  "Great! You and the model both got the correct answer"
-    elif userclass == classes[int(truth)]:
-        return "Great! You guessed it right"
-    elif prediction == classes[int(truth)]:
-        return "The AI model got it right this time, try again!"
+def compare_score(userclass, truth, has_guessed):
+    if userclass is None:
+        return"Try guessing a category!"
+    else:
+        if userclass == classes[int(truth)]:
+            return "Great! You guessed it right"
+        else:
+            return "The right answer was " +str(classes[int(truth)])+ "! Try guessing the next image."
 
 with gr.Blocks() as demo:
     user_score = gr.State(0)
@@ -80,7 +80,9 @@ with gr.Blocks() as demo:
     has_guessed = gr.State(False)
 
     gr.Markdown("# ImageNet Quiz")
-    gr.Markdown("Try guessing the category of each image displayed, from the options provided below.")
+    gr.Markdown("### ImageNet is one of the most popular datasets used for training and evaluating AI models.")
+    gr.Markdown("### But many of its categories are hard to guess, even for humans.")
+    gr.Markdown("#### Try your hand at guessing the category of each image displayed, from the options provided. Compare your answers to that of a neural network trained on the dataset, and see if you can do better!")
     with gr.Row():
 
         with gr.Column(min_width= 900):
@@ -89,14 +91,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             prediction = gr.Label(label="The AI model predicts:")
             score = gr.Label(label="Your Score")
-            message = gr.Text(label="Who guessed it right?")
+            message = gr.Label(label="Did you guess it right?")
 
     btn = gr.Button("Next image")
 
     demo.load(random_image, None, [image, image_label, radio, prediction])
     radio.change(model_classify, [radio, image], [prediction, model_class, has_guessed])
     radio.change(check_score, [radio, image_label, user_score, total_score, has_guessed], [user_score, score, total_score])
-    radio.change(compare_score, [radio, prediction, image_label, has_guessed], message)
+    radio.change(compare_score, [radio, image_label, has_guessed], message)
     btn.click(random_image, None, [image, image_label, radio, prediction])
     btn.click(lambda :False, None, has_guessed)