Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -139,10 +139,10 @@ def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolu
|
|
| 139 |
).frames[0]
|
| 140 |
# Export to video
|
| 141 |
video_path = "output.mp4"
|
| 142 |
-
video_bytes = io.BytesIO()
|
| 143 |
-
export_to_video(video,
|
| 144 |
-
|
| 145 |
-
return
|
| 146 |
|
| 147 |
@torch.inference_mode()
|
| 148 |
def call_pipe(
|
|
|
|
| 139 |
).frames[0]
|
| 140 |
# Export to video
|
| 141 |
video_path = "output.mp4"
|
| 142 |
+
# video_bytes = io.BytesIO()
|
| 143 |
+
export_to_video(video, video_path, fps=24)
|
| 144 |
+
torch.cuda.empty_cache()
|
| 145 |
+
return video_path
|
| 146 |
|
| 147 |
@torch.inference_mode()
|
| 148 |
def call_pipe(
|