SajjadAyoubi commited on
Commit
d3e4cd0
·
1 Parent(s): 665ec8e

Update app.py

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