Update app.py

app.py

@@ -176,9 +176,9 @@ def Visual_re_ranker(caption_G, caption_B, caption_VR, visual_context_label, vis
     sim_3 = get_sim(sim_3)
  
 
-    LM_1 = cloze_prob(caption_G)
-    LM_2 = cloze_prob(caption_B)
-    LM_3 = cloze_prob(caption_VR)
+    LM_1 = sentence_prob_mean(caption_G)
+    LM_2 = sentence_prob_mean(caption_B)
+    LM_3 = sentence_prob_mean(caption_VR)
 
     #LM  = scorer.sentence_score(caption, reduce="mean")
     score_1 = pow(float(LM_1),pow((1-float(sim_1))/(1+ float(sim_1)),1-float(visual_context_prob)))