Spaces:

AhmedSSabir
/

demo-for-Visual-Re-ranker

Sleeping

AhmedSSabir commited on Apr 19

Commit

6f13d63

•

1 Parent(s): 40180ec

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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  = 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)))

     sim_3 = get_sim(sim_3)
+    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)))