AhmedSSabir commited on
Commit
c673cdb
1 Parent(s): 59c43da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -202,8 +202,8 @@ def Visual_re_ranker(caption_man, caption_woman, context_label, context_prob):
202
  sim_w = get_sim(sim_w)
203
 
204
 
205
- LM_man = cloze_prob(caption_man)
206
- LM_woman = cloze_prob(caption_woman)
207
  #LM = scorer.sentence_score(caption, reduce="mean")
208
  score_man = pow(float(LM_man),pow((1-float(sim_m))/(1+ float(sim_m)),1-float(context_prob)))
209
  score_woman = pow(float(LM_woman),pow((1-float(sim_w))/(1+ float(sim_w)),1-float(context_prob)))
 
202
  sim_w = get_sim(sim_w)
203
 
204
 
205
+ LM_man = sentence_prob_mean(caption_man)
206
+ LM_woman = sentence_prob_mean(caption_woman)
207
  #LM = scorer.sentence_score(caption, reduce="mean")
208
  score_man = pow(float(LM_man),pow((1-float(sim_m))/(1+ float(sim_m)),1-float(context_prob)))
209
  score_woman = pow(float(LM_woman),pow((1-float(sim_w))/(1+ float(sim_w)),1-float(context_prob)))