Update app.py

app.py

@@ -4,17 +4,18 @@ from diffusers import CogVideoXImageToVideoPipeline
 from diffusers.utils import export_to_video, load_image
 import gradio as gr
 
-@spaces.GPU(duration=250)
-def generate_video(prompt, image):
-    # Carregar a pipeline pré-treinada
-    pipe = CogVideoXImageToVideoPipeline.from_pretrained(
+pipe = CogVideoXImageToVideoPipeline.from_pretrained(
         "THUDM/CogVideoX-5b-I2V",
         torch_dtype=torch.bfloat16
     )
     
-    pipe.enable_sequential_cpu_offload()
-    pipe.vae.enable_tiling()
-    pipe.vae.enable_slicing()
+pipe.enable_sequential_cpu_offload()
+pipe.vae.enable_tiling()
+pipe.vae.enable_slicing()
+
+@spaces.GPU(duration=250)
+def generate_video(prompt, image):
+    pipe.to("cuda")
     
     video = pipe(
         prompt=prompt,