Update app.py

app.py

@@ -87,14 +87,14 @@ def cos_sim(a, b):
 
 
 
-def Visual_re_ranker(caption_man, caption_woman, context_label, context_prob):
+def Visual_re_ranker(caption_man, caption_woman, visual_context_label, context_prob):
     caption_man = caption_man  
     caption_woman = caption_woman
-    context_label= context_label
+    visual_context_label = visual_context_label
     context_prob = context_prob
     caption_emb_man = model_sts.encode(caption_man, convert_to_tensor=True)
     caption_emb_woman = model_sts.encode(caption_woman, convert_to_tensor=True)
-    context_label_emb = model_sts.encode(context_label, convert_to_tensor=True)
+    context_label_emb = model_sts.encode(visual_context_label, convert_to_tensor=True)
 
     sim_m =  cosine_scores = util.pytorch_cos_sim(caption_emb_man, context_label_emb)
     sim_m = sim_m.cpu().numpy()