ia

app.py

@@ -11,7 +11,7 @@ from camera_movement_estimator import CameraMovementEstimator
 from view_transformer import ViewTransformer
 from speed_and_distance_estimator import SpeedAndDistance_Estimator
 
-def process_video(input_video, player_stats, ball_stats):
+def process_video(input_video, player_stats=True, ball_stats=True):
     print(input_video)
     # Read Video
     video_frames = read_video(input_video)  # use the uploaded file
@@ -19,7 +19,7 @@ def process_video(input_video, player_stats, ball_stats):
     # Initialize Tracker
     tracker = Tracker('models/best.pt')
 
-    if input_video == "input_videos/121364_0_small.mp4":
+    if input_video.contains("121364_0_small.mp4"):
         print("loading cached tracks")
         tracks = tracker.get_object_tracks(video_frames,
                                        read_from_stub=True,
@@ -35,7 +35,7 @@ def process_video(input_video, player_stats, ball_stats):
 
     # Camera movement estimator
     camera_movement_estimator = CameraMovementEstimator(video_frames[0])
-    if input_video == "input_videos/121364_0_small.mp4":
+    if input_video.contains("121364_0_small.mp4"):
         print("loading cached camera movements")
         camera_movement_per_frame = camera_movement_estimator.get_camera_movement(video_frames,
                                                                                 read_from_stub=True,

requirements.txt

@@ -1,3 +1,4 @@
+gradio==2.9.1
 opencv-contrib-python==4.10.0.84
 opencv-python==4.10.0.84
 opencv-python-headless==4.10.0.84
@@ -10,4 +11,4 @@ matplotlib==3.8.0
 matplotlib-inline==0.1.7
 matplotlib-venn==1.1.1
 pandas==2.2.2
-sklearn-pandas==2.2.0
+sklearn-pandas==2.2.0