Update app.py

app.py

@@ -37,7 +37,9 @@ 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]
+        return model.config.id2label[predicted_class_idx], True
+    else:
+        return None, False
 
 def random_image():
     imname = random.choice(images)
@@ -51,13 +53,18 @@ def random_image():
     options = [classes[int(i)] for i in options]
     return im, label, gr.Radio.update(value=None, choices=options), None
 
-def check_score(pred, truth, current_score, total_score):
-    if pred == classes[int(truth)]:
-        total_score +=1
-        return current_score + 1, f"Your score is {current_score+1} out of {total_score}"
+def check_score(pred, truth, current_score, total_score, has_guessed):
+    if not(has_guessed):
+        if pred == "lion":
+            total_score +=1
+            return current_score + 1, f"Your score is {current_score+1} out of {total_score}", total_score
+        else:
+            if pred is not None:
+                total_score +=1
+            return current_score, f"Your score is {current_score} out of {total_score}", total_score
     else:
-        total_score +=1
-        return current_score, f"Your score is {current_score} out of {total_score}"
+        return current_score, f"Your score is {current_score} out of {total_score}", total_score
+        
 
 
 def compare_score(userclass, prediction):
@@ -71,6 +78,7 @@ with gr.Blocks() as demo:
     image_label = gr.State()
     prediction = gr.State()
     total_score = gr.State(0)
+    has_guessed = gr.State(False)
 
     with gr.Row():
         with gr.Column(min_width= 900):
@@ -84,10 +92,11 @@ with gr.Blocks() as demo:
     btn = gr.Button("Next image")
 
     demo.load(random_image, None, [image, image_label, radio, prediction])
-    radio.change(model_classify, [radio, image], prediction)
-    radio.change(check_score, [radio, image_label, user_score, total_score], [user_score, score])
+    radio.change(model_classify, [radio, image], [prediction, 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], message)
     btn.click(random_image, None, [image, image_label, radio, prediction])
+    btn.click(lambda :False, None, has_guessed)
 
 
 demo.launch()