Update app.py

app.py

@@ -7,12 +7,6 @@ from moviepy.editor import *
 token = os.environ.get('HF_TOKEN')
 pix2pix = gr.Blocks.load(name="spaces/fffiloni/instruct-pix2pix-clone", api_key=token)
 
-def trim_video(video_in, trim_value):
-    clip = VideoFileClip(video_in)
-    clip1 = clip.subclip((0,00),(trim_value,00))
-    clip1.write_videofile('edited.mp4',codec='libx264')
-    return 'edited.mp4'
-
 def get_frames(video_in):
     frames = []
     # Opens the Video file
@@ -41,17 +35,19 @@ def create_video(frames, fps):
 
 
 def infer(prompt,video_in, seed_in, trim_value):
-    trimmed_vid = trim_video(video_in, trim_value)
-    break_vid = get_frames(trimmed_vid)
+    
+    break_vid = get_frames(video_in)
     
     frames_list= break_vid[0]
     fps = break_vid[1]
+    n_frame = trim_value*fps
     result_frames = []
     
-    for i in frames_list:
+    for i in frames_list[0:n_frame]:
         pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in,fn_index=0)
         images = [os.path.join(pix2pix_img[0], img) for img in os.listdir(pix2pix_img[0])]
         result_frames.append(images[0])
+        print("frame " + i + ": done;")
 
     final_vid = create_video(result_frames, fps)
     
@@ -68,7 +64,7 @@ with gr.Blocks(css='style.css') as demo:
             with gr.Column():
                 with gr.Row():
                     seed_inp = gr.Slider(minimum=0, maximum=10000, step=1, value=123456)
-                    trim_in = gr.Slider(label="video max duration (seconds)", minimun=2, maximum=10, step=1, value=2)
+                    trim_in = gr.Slider(label="Cut video at (s)", minimun=2, maximum=10, step=1, value=2)
                 video_out = gr.Video(label="Pix2pix video result")
                 submit_btn = gr.Button("Generate Pix2Pix video")