Spaces:
Runtime error
Runtime 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,
|