Update scripts/gradio/i2v_test_application.py

scripts/gradio/i2v_test_application.py

@@ -12,6 +12,7 @@ from einops import rearrange
 from cldm.model import load_state_dict
 import cv2
 
+import spaces 
 
 def extract_frames(video_path):
     # 動画ファイルを読み込む
@@ -77,6 +78,7 @@ class Image2Video():
         self.model_list = model_list
         self.save_fps = 8
 
+    @spaces.GPU(duration=100)
     def get_image(self, image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, image2=None, frame_guides=None,control_scale=0.6):
         control_frames = extract_frames(frame_guides)
         seed_everything(seed)