Update gradio_app.py

gradio_app.py

@@ -64,7 +64,7 @@ def extract_frames(video_path):
     # 動画ファイルを読み込む
     cap = cv2.VideoCapture(video_path)
     
-    frame_list = []
+    frame_list = [] 
     frame_num = 0
     
     while True:
@@ -238,8 +238,8 @@ def get_image(image, sketch, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, see
         batch_samples = batch_ddim_sampling(model, cond, noise_shape, n_samples=1, ddim_steps=steps, ddim_eta=eta, cfg_scale=cfg_scale, hs=hs)
 
         ## remove the last frame
-        if image2 is None:
-            batch_samples = batch_samples[:,:,:,:-1,...]
+        # if image2 is None:
+        batch_samples = batch_samples[:,:,:,:-1,...]
         ## b,samples,c,t,h,w
         prompt_str = prompt.replace("/", "_slash_") if "/" in prompt else prompt
         prompt_str = prompt_str.replace(" ", "_") if " " in prompt else prompt_str