SajjadAyoubi commited on
Commit
7a4d3b4
·
1 Parent(s): 4fcb178

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def load():
12
  text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
13
  tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
14
  links = np.load('data.npy', allow_pickle=True)
15
- image_embeddings = torch.load('embeddings.pt')
16
  return text_encoder, tokenizer, links, image_embeddings
17
 
18
 
 
12
  text_encoder = RobertaModel.from_pretrained('SajjadAyoubi/clip-fa-text')
13
  tokenizer = AutoTokenizer.from_pretrained('SajjadAyoubi/clip-fa-text')
14
  links = np.load('data.npy', allow_pickle=True)
15
+ image_embeddings = torch.load('embedding.pt')
16
  return text_encoder, tokenizer, links, image_embeddings
17
 
18