added threads

main_noweb.py

@@ -43,49 +43,13 @@ hand = MMPoseInferencer("hand", device=device)
 
 print("[INFO]: Downloaded models!")
 
-def check_fps(video):
-    cap = cv2.VideoCapture(video)
-    nframes = cap.get(cv2.CAP_PROP_FRAME_COUNT) 
-    fps = cap.get(cv2.CAP_PROP_FPS)   
-    return nframes, fps
-
-
-def get_frames(video, fps=50, height=512, width=512):
-    
-    clip = moviepy.VideoFileClip(video)
-    print(clip.duration)
-    if clip.duration > 10:
-            raise gr.Error("Please provide or record a video shorter than 10 seconds...")
-    
-    split_tup = os.path.splitext(video)
-
-    file_name = split_tup[0]
-    file_extension = split_tup[1]
-
-    if file_extension != ".mp4":
-        print("Converting to mp4")
-        video = file_name+".mp4"
-
-    #check fps
-    if clip.fps > fps:
-        print("vide rate is over 15, resetting to 15")
-        #clip_resized = moviepy.clip.resize(height=height)
-        clip.write_videofile(video, fps=fps)
-    else:
-        print("video rate is OK")
-        #clip_resized = clip.resize(height=height)
-        #clip.write_videofile(video, fps=clip.fps)
-    return video
-    
-
 def check_extension(video):
     
     clip = moviepy.VideoFileClip(video)
-    print(clip.duration)
+
     if clip.duration > 10:
             raise gr.Error("Please provide or record a video shorter than 10 seconds...")
 
-
     split_tup = os.path.splitext(video)
 
     # extract the file name and extension
@@ -96,7 +60,7 @@ def check_extension(video):
         print("Converting to mp4")
 
         video = file_name+".mp4"
-        clip.write_videofile(video)
+        clip.write_videofile(video, threads = 8)
     
     return video