Update gradio_app.py

gradio_app.py

@@ -51,8 +51,9 @@ def extract_frames(video_path):
     
     return frame_list
 
-@spaces.GPU(duration=30)
+
 class Image2Video():
+    @spaces.GPU
     def __init__(self,result_dir='./tmp/',gpu_num=1,resolution='256_256') -> None:
         self.resolution = (int(resolution.split('_')[0]), int(resolution.split('_')[1])) #hw
         self.download_model()
@@ -97,6 +98,7 @@ class Image2Video():
         print("init done.")
 
     # @spaces.GPU(duration=100)
+    @spaces.GPU
     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):
         print("enter fn")
         control_frames = extract_frames(frame_guides)