Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ unique_pairs = [(item[0], item[1]) for item in training_pairs if item[0] not in
|
|
| 25 |
coco_images, _ = zip(*unique_pairs)
|
| 26 |
|
| 27 |
# Load model (update)
|
| 28 |
-
model = OneEncoder.from_pretrained("bilalfaye/OneEncoder-
|
| 29 |
|
| 30 |
# Load coco image features
|
| 31 |
coco_image_features = torch.load("image_embeddings_best.pt", map_location=CFG.device)
|
|
|
|
| 25 |
coco_images, _ = zip(*unique_pairs)
|
| 26 |
|
| 27 |
# Load model (update)
|
| 28 |
+
model = OneEncoder.from_pretrained("bilalfaye/OneEncoder-text-image-audio")
|
| 29 |
|
| 30 |
# Load coco image features
|
| 31 |
coco_image_features = torch.load("image_embeddings_best.pt", map_location=CFG.device)
|