updates

app.py

@@ -135,8 +135,8 @@ def save_annotation(caption, session_data):
         dataset.push_to_hub(dataset_name)
         print("Pushed updated dataset")
 
-        # Clear user's current image if the validation image has been annotated five times
-        if ("val" not in image_id) or (annotation_count > 2):
+        # Clear user's current image if the validation image has been annotated twice
+        if ("val" not in image_id) or (annotation_count >= 2):
             session_data["current_image"] = None
 
     # Fetch the next image