Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -125,8 +125,9 @@ def resize_image_to_bucket(image: Union[Image.Image, np.ndarray], bucket_reso: T
|
|
| 125 |
image = image[crop_top:crop_top + bucket_height, crop_left:crop_left + bucket_width]
|
| 126 |
return image
|
| 127 |
|
| 128 |
-
#
|
| 129 |
-
@torch.inference_mode()
|
|
|
|
| 130 |
def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
|
| 131 |
# Debugging print statements
|
| 132 |
print(f"Frame 1 Type: {type(frame1)}")
|
|
|
|
| 125 |
image = image[crop_top:crop_top + bucket_height, crop_left:crop_left + bucket_width]
|
| 126 |
return image
|
| 127 |
|
| 128 |
+
#
|
| 129 |
+
# @torch.inference_mode()
|
| 130 |
+
@spaces.GPU(duration=120)
|
| 131 |
def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
|
| 132 |
# Debugging print statements
|
| 133 |
print(f"Frame 1 Type: {type(frame1)}")
|