Spaces:
Running
Running
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)
|