AhmedSSabir commited on
Commit
459ca58
1 Parent(s): c1e19ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def Visual_re_ranker(caption, visual_context_label, visual_context_prob):
108
  caption = caption
109
  visual_context_label= visual_context_label
110
  visual_context_prob = visual_context_prob
111
- caption_emb = model.encode(caption, convert_to_tensor=True)
112
  visual_context_label_emb = model_1.encode(visual_context_label, convert_to_tensor=True)
113
 
114
 
 
108
  caption = caption
109
  visual_context_label= visual_context_label
110
  visual_context_prob = visual_context_prob
111
+ caption_emb = model_1.encode(caption, convert_to_tensor=True)
112
  visual_context_label_emb = model_1.encode(visual_context_label, convert_to_tensor=True)
113
 
114