Update app.py
Browse files
app.py
CHANGED
|
@@ -145,11 +145,11 @@ def Visual_re_ranker(caption_man, caption_woman, context_label, context_prob):
|
|
| 145 |
sim_w = get_sim(sim_w)
|
| 146 |
|
| 147 |
|
| 148 |
-
|
| 149 |
-
|
| 150 |
|
| 151 |
-
LM_man = cloze_prob(caption_man)
|
| 152 |
-
LM_woman = cloze_prob(caption_woman)
|
| 153 |
|
| 154 |
|
| 155 |
|
|
|
|
| 145 |
sim_w = get_sim(sim_w)
|
| 146 |
|
| 147 |
|
| 148 |
+
LM_man = sentence_prob_mean(caption_man)
|
| 149 |
+
LM_woman = sentence_prob_mean(caption_woman)
|
| 150 |
|
| 151 |
+
#LM_man = cloze_prob(caption_man)
|
| 152 |
+
#LM_woman = cloze_prob(caption_woman)
|
| 153 |
|
| 154 |
|
| 155 |
|