Ana Sanchez commited on
Commit
21dd8b4
·
1 Parent(s): 74b3cd2

Update app

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -440,6 +440,7 @@ def images_from_molecule():
440
  "How many images would you like to be retrieve?",
441
  ("5", "10", "20")
442
  )
 
443
 
444
  img_features_torch = torch.load(image_features, map_location=device)
445
  img_features = img_features_torch["img_features"]
 
440
  "How many images would you like to be retrieve?",
441
  ("5", "10", "20")
442
  )
443
+ top_n = int(top_n)
444
 
445
  img_features_torch = torch.load(image_features, map_location=device)
446
  img_features = img_features_torch["img_features"]