helloworld-S commited on
Commit
ea7754d
·
verified ·
1 Parent(s): a639827

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -164,7 +164,7 @@ def generate_image(
164
  *images_captions_faces, # Combine all unpacked arguments into one tuple
165
  ):
166
  torch.cuda.empty_cache()
167
- num_images = 4
168
 
169
  # Determine the number of images, captions, and faces based on the indexs length
170
  images = list(images_captions_faces[:num_inputs])
 
164
  *images_captions_faces, # Combine all unpacked arguments into one tuple
165
  ):
166
  torch.cuda.empty_cache()
167
+ num_images = 1
168
 
169
  # Determine the number of images, captions, and faces based on the indexs length
170
  images = list(images_captions_faces[:num_inputs])