AhmedSSabir commited on
Commit
3e97a0d
1 Parent(s): 5b61e93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -162,7 +162,7 @@ def Visual_re_ranker(sentence_man, sentence_woman, context_label, context_prob):
162
  context_prob = context_prob
163
  sentence_emb_man = model_sbert.encode(sentence_man, convert_to_tensor=True)
164
  sentence_emb_woman = model_sbert.encode(sentence_woman, convert_to_tensor=True)
165
- visual_context_label_emb = model_sbert.encode(context_label, convert_to_tensor=True)
166
 
167
  sim_m = cosine_scores = util.pytorch_cos_sim(sentence_emb_man, context_label_emb)
168
  sim_m = sim_m.cpu().numpy()
 
162
  context_prob = context_prob
163
  sentence_emb_man = model_sbert.encode(sentence_man, convert_to_tensor=True)
164
  sentence_emb_woman = model_sbert.encode(sentence_woman, convert_to_tensor=True)
165
+ context_label_emb = model_sbert.encode(context_label, convert_to_tensor=True)
166
 
167
  sim_m = cosine_scores = util.pytorch_cos_sim(sentence_emb_man, context_label_emb)
168
  sim_m = sim_m.cpu().numpy()