Update app.py

app.py

@@ -109,22 +109,18 @@ def infer(video_frames, masks_frames, project_name):
 def get_frames(video_in, img_type):
     frames = []
     #resize the video
-    # Cut the video to the first 5 seconds
-    #video_out = f"{img_type}_video_cut.mp4"
-    
-    #ffmpeg_extract_subclip(video_in, t1=0, t2=3, targetname=video_out)
-    # Now, proceed with resizing the cut video
+
     clip = VideoFileClip(video_in)
     
     #check fps
     
     if clip.fps > 30:
         print("vide rate is over 30, resetting to 30")
-        clip_resized = clip.resize(width=512)
+        clip_resized = clip.resize(height=512)
         clip_resized.write_videofile(f"{img_type}_video_resized.mp4", fps=30)
     else:
         print("video rate is OK")
-        clip_resized = clip.resize(width=512)
+        clip_resized = clip.resize(height=512)
         clip_resized.write_videofile(f"{img_type}_video_resized.mp4", fps=clip.fps)
     
     
@@ -175,7 +171,10 @@ def infer_auto(project_name, video_in, subject_to_remove):
     print(video_in)
     matte_video = get_matte(video_in, subject_to_remove)
 
-    video_frames = get_frames(video_in, "source")
+    # Cut the video to the first 3 seconds
+    video_cut = f"video_cut.mp4"
+    ffmpeg_extract_subclip(video_in, t1=0, t2=3, targetname=video_cut)
+    video_frames = get_frames(video_cut, "source")
     print(video_frames[0])
  
     masks_frames = get_frames(matte_video, "mask")