SajjadAyoubi commited on
Commit
7d696ff
·
1 Parent(s): 55b4896

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -14,9 +14,10 @@ def load():
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 model, processor, df, embeddings
18
 
19
- model, processor, df, embeddings = load()
 
20
 
21
 
22
  def get_html(url_list):
 
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
 
19
+ tokenizer, text_encoder, links, image_embeddings = load()
20
+
21
 
22
 
23
  def get_html(url_list):