Kvikontent commited on
Commit
edbc6e4
1 Parent(s): f68c811

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ def generate_images(prompt, num_images):
12
  """Generates images using the image pipeline."""
13
  images = []
14
  for _ in range(num_images):
15
- generated_image = image_pipeline.pipe(prompt=prompt).images[0]
16
  images.append(generated_image)
17
  return images
18
 
@@ -22,7 +22,7 @@ def generate_videos(images):
22
  videos = []
23
  for image in images:
24
  # Wrap the image in a list as expected by the pipeline
25
- generated_video = video_pipeline.pipe(images=[image]).images[0]
26
  videos.append(generated_video)
27
  return videos
28
 
 
12
  """Generates images using the image pipeline."""
13
  images = []
14
  for _ in range(num_images):
15
+ generated_image = image_pipeline(prompt=prompt).images[0]
16
  images.append(generated_image)
17
  return images
18
 
 
22
  videos = []
23
  for image in images:
24
  # Wrap the image in a list as expected by the pipeline
25
+ generated_video = video_pipeline(images=[image]).images[0]
26
  videos.append(generated_video)
27
  return videos
28