Nikolaygod777 commited on
Commit
f3d7684
·
verified ·
1 Parent(s): 4311c66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -44,7 +44,8 @@ def generate_image(prompt):
44
  # Генерация видео
45
  def generate_video(image_path):
46
  image = Image.open(image_path)
47
- video_frames = video_pipe(image, num_inference_steps=10, num_frames=16).frames
 
48
  video_path = "generated_video.mp4"
49
  video_pipe.save_video(video_frames, video_path, fps=8)
50
  return video_path
 
44
  # Генерация видео
45
  def generate_video(image_path):
46
  image = Image.open(image_path)
47
+ with autocast(device):
48
+ video_frames = video_pipe(image, num_inference_steps=10, num_frames=16).frames
49
  video_path = "generated_video.mp4"
50
  video_pipe.save_video(video_frames, video_path, fps=8)
51
  return video_path