Spaces:
Running
Running
Ana Sanchez commited on
Commit ·
21dd8b4
1
Parent(s): 74b3cd2
Update app
Browse files
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"]
|