benibraz commited on
Commit
520e8b2
1 Parent(s): 623c312
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -201,7 +201,7 @@ def generate_video_from_text(
201
 
202
  images = pipeline(
203
  num_inference_steps=num_inference_steps,
204
- num_images_per_prompt=num_images_per_prompt,
205
  guidance_scale=guidance_scale,
206
  generator=generator,
207
  output_type="pt",
@@ -218,6 +218,7 @@ def generate_video_from_text(
218
  ).images
219
 
220
  output_path = tempfile.mktemp(suffix=".mp4")
 
221
  video_np = images.squeeze(0).permute(1, 2, 3, 0).cpu().float().numpy()
222
  video_np = (video_np * 255).astype(np.uint8)
223
  height, width = video_np.shape[1:3]
@@ -385,6 +386,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
385
  - Describe lighting and colors
386
  - Note any changes or sudden events
387
 
 
 
388
  🎮 Parameter Guide
389
 
390
  - Resolution Preset: Higher resolutions for detailed scenes, lower for faster generation and simpler scenes
 
201
 
202
  images = pipeline(
203
  num_inference_steps=num_inference_steps,
204
+ num_images_per_prompt=1,
205
  guidance_scale=guidance_scale,
206
  generator=generator,
207
  output_type="pt",
 
218
  ).images
219
 
220
  output_path = tempfile.mktemp(suffix=".mp4")
221
+ print(images.shape)
222
  video_np = images.squeeze(0).permute(1, 2, 3, 0).cpu().float().numpy()
223
  video_np = (video_np * 255).astype(np.uint8)
224
  height, width = video_np.shape[1:3]
 
386
  - Describe lighting and colors
387
  - Note any changes or sudden events
388
 
389
+ See examples for more inspiration.
390
+
391
  🎮 Parameter Guide
392
 
393
  - Resolution Preset: Higher resolutions for detailed scenes, lower for faster generation and simpler scenes