Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -136,10 +136,11 @@ def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolu
|
|
| 136 |
).frames[0]
|
| 137 |
# Export to video
|
| 138 |
video_path = "output.mp4"
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
|
| 142 |
return video_bytes
|
|
|
|
| 143 |
@torch.inference_mode()
|
| 144 |
def call_pipe(
|
| 145 |
pipe,
|
|
|
|
| 136 |
).frames[0]
|
| 137 |
# Export to video
|
| 138 |
video_path = "output.mp4"
|
| 139 |
+
video_bytes = io.BytesIO()
|
| 140 |
+
export_to_video(video, video_bytes, fps=24)
|
| 141 |
+
video_bytes.seek(0) # Reset the file pointer to the beginning
|
| 142 |
return video_bytes
|
| 143 |
+
|
| 144 |
@torch.inference_mode()
|
| 145 |
def call_pipe(
|
| 146 |
pipe,
|