SeemG commited on
Commit
be8c2fb
·
verified ·
1 Parent(s): e4353e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ _, valid_df = make_train_valid_dfs()
39
  print("Going to invoke make_train_valid_dfs")
40
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
41
 
42
- def generate_images(text, num_images=6):
43
  generated_images = clip_image_search(model,
44
  image_embeddings,
45
  text,
 
39
  print("Going to invoke make_train_valid_dfs")
40
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
41
 
42
+ def generate_images(text, num_images=16):
43
  generated_images = clip_image_search(model,
44
  image_embeddings,
45
  text,