Update app.py

app.py

@@ -89,10 +89,9 @@ def unified_prediction_function(frame):
     if frame is None:
         return None, create_prediction_html({})
     
-    # The predictor class handles all the complex ML logic
-    annotated_frame, probabilities = predictor.process_frame(frame)
-    
     annotated_frame, probabilities = predictor.process_frame_for_upload(frame)
+    
+    return annotated_frame, create_prediction_html(probabilities)
 
 def process_video(video_path, progress=gr.Progress(track_tqdm=True)):
     """Processes an uploaded video file frame-by-frame."""