Update app.py

app.py

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