SajjadAyoubi commited on
Commit
f0fa5d9
·
1 Parent(s): 13213b8

Update app.py

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