AhmedSSabir
commited on
Commit
·
1e4c550
1
Parent(s):
4387c41
Update app.py
Browse files
app.py
CHANGED
@@ -145,11 +145,11 @@ def Visual_re_ranker(caption_G, caption_B, caption_VR, visual_context_label, vis
|
|
145 |
score_3 = pow(float(LM_3),pow((1-float(sim_3))/(1+ float(sim_3)),1-float(visual_context_prob)))
|
146 |
|
147 |
#return {"LM": float(LM)/1, "sim": float(sim)/1, "score": float(score)/1 }
|
148 |
-
return {"
|
149 |
#return LM, sim, score
|
150 |
|
151 |
|
152 |
-
|
153 |
|
154 |
demo = gr.Interface(
|
155 |
fn=Visual_re_ranker,
|
|
|
145 |
score_3 = pow(float(LM_3),pow((1-float(sim_3))/(1+ float(sim_3)),1-float(visual_context_prob)))
|
146 |
|
147 |
#return {"LM": float(LM)/1, "sim": float(sim)/1, "score": float(score)/1 }
|
148 |
+
return {"Greedy": float(score_1)/1, "Best-Beam-5": float(score_2)/1, "Visual_re-Ranker": float(score_3)/1 }
|
149 |
#return LM, sim, score
|
150 |
|
151 |
|
152 |
+
|
153 |
|
154 |
demo = gr.Interface(
|
155 |
fn=Visual_re_ranker,
|