Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import safetensors.torch
|
3 |
import torchvision.transforms.v2 as transforms
|
@@ -99,7 +100,7 @@ def resize_image_to_bucket(image: Union[Image.Image, np.ndarray], bucket_reso: T
|
|
99 |
return image
|
100 |
|
101 |
|
102 |
-
|
103 |
def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
|
104 |
# Debugging print statements
|
105 |
print(f"Frame 1 Type: {type(frame1)}")
|
|
|
1 |
+
import spaces
|
2 |
import gradio as gr
|
3 |
import safetensors.torch
|
4 |
import torchvision.transforms.v2 as transforms
|
|
|
100 |
return image
|
101 |
|
102 |
|
103 |
+
@spaces.GPU()
|
104 |
def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
|
105 |
# Debugging print statements
|
106 |
print(f"Frame 1 Type: {type(frame1)}")
|