bilalfaye commited on
Commit
990c0d6
·
verified ·
1 Parent(s): 8588038

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-texte-image-audio")
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)