Update app.py

app.py

@@ -48,9 +48,11 @@ def infer(prompt,video_in, seed_in, trim_value):
     frames_list= break_vid[0]
     fps = break_vid[1]
     n_frame = int(trim_value*fps)
+    if n_frame >= len(frames_list):
+        n_frame = len(frames_list)
     result_frames = []
     print("set stop frames to: " + n_frame)
-    for i in frames_list[0:n_frame]:
+    for i in frames_list[0:int(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])