evelyncsb commited on
Commit
18cc9d1
1 Parent(s): 2f24ca3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def generate_image(text):
35
 
36
  similarity = text_features.cpu().numpy() @ image_features.cpu().numpy().T
37
 
38
- #pega index maior
39
  index_img = np.argmax(similarity)
40
  img_name = os.path.basename(image_paths[index_img])
41
  im = Image.open(f"./assets/images/{img_name}").convert("RGB")
 
35
 
36
  similarity = text_features.cpu().numpy() @ image_features.cpu().numpy().T
37
 
38
+
39
  index_img = np.argmax(similarity)
40
  img_name = os.path.basename(image_paths[index_img])
41
  im = Image.open(f"./assets/images/{img_name}").convert("RGB")