pm6six commited on
Commit
62ad765
1 Parent(s): 6641dc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -40,5 +40,15 @@ if pipeline_available:
40
  pipe.vae.enable_tiling()
41
  pipe.vae.enable_slicing()
42
 
 
43
  # Generate video
44
- with st.spinner("Generating video
 
 
 
 
 
 
 
 
 
 
40
  pipe.vae.enable_tiling()
41
  pipe.vae.enable_slicing()
42
 
43
+
44
  # Generate video
45
+ with st.spinner("Generating video... This may take a while."):
46
+ video_frames = pipe(
47
+ prompt=prompt,
48
+ image=image,
49
+ num_videos_per_prompt=1,
50
+ num_inference_steps=50,
51
+ num_frames=81,
52
+ guidance_scale=6,
53
+ generator=torch.Generator(device=device).manual_seed(42),
54
+ ).frames[0]