AhmedSSabir
commited on
Commit
•
459ca58
1
Parent(s):
c1e19ca
Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ def Visual_re_ranker(caption, visual_context_label, visual_context_prob):
|
|
108 |
caption = caption
|
109 |
visual_context_label= visual_context_label
|
110 |
visual_context_prob = visual_context_prob
|
111 |
-
caption_emb =
|
112 |
visual_context_label_emb = model_1.encode(visual_context_label, convert_to_tensor=True)
|
113 |
|
114 |
|
|
|
108 |
caption = caption
|
109 |
visual_context_label= visual_context_label
|
110 |
visual_context_prob = visual_context_prob
|
111 |
+
caption_emb = model_1.encode(caption, convert_to_tensor=True)
|
112 |
visual_context_label_emb = model_1.encode(visual_context_label, convert_to_tensor=True)
|
113 |
|
114 |
|