LPX55 commited on
Commit
216eb04
·
verified ·
1 Parent(s): 76f7da1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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, video_bytes, fps=24)
144
- video_bytes.seek(0) # Reset the file pointer to the beginning
145
- return video_bytes
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(