Spaces:
Running
on
Zero
Running
on
Zero
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(
|