Update app.py

app.py

@@ -21,13 +21,18 @@ for model_id in model_ids:
 
 
 def get_frame_count(filepath):
-    video = cv2.VideoCapture(filepath) 
-    frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
-    video.release()
+    if filepath is not None:
+        video = cv2.VideoCapture(filepath) 
+        frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
+    
+        video.release()
 
-    #LIMITS
-    if frame_count > 24 :
-        frame_count = 24 # limit to 24 frames to avoid cuDNN errors
+        #LIMITS
+        if frame_count > 24 :
+            frame_count = 24 # limit to 24 frames to avoid cuDNN errors
+
+    else:
+        frame_count = 1
     return gr.update(maximum=frame_count)
 
 def get_video_dimension(filepath):