Spaces:
Sleeping
Sleeping
Prepare for choices chage
Browse files
app.py
CHANGED
@@ -1029,9 +1029,9 @@ max_image_choices = 10 # Currently up to 30 images but relevance degrades quick
|
|
1029 |
no_image_idxs = [i for i, imgs in enumerate(babelnet_images) if len(imgs) == 0]
|
1030 |
IMG_HEIGHT, IMG_WIDTH = 512, 512
|
1031 |
|
1032 |
-
precomputed_results = None
|
1033 |
-
if os.path.exists("data/precomputed_results.json"):
|
1034 |
-
|
1035 |
|
1036 |
request_header = {
|
1037 |
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36"
|
@@ -1129,7 +1129,7 @@ def prepare(raw_idx, text_embeddings, class_order):
|
|
1129 |
min(len(babelnet_images[class_idx]), max_image_choices)
|
1130 |
)
|
1131 |
img_url = babelnet_images[class_idx][img_idx]["url"]
|
1132 |
-
class_labels =
|
1133 |
|
1134 |
if not precomputed_results:
|
1135 |
try:
|
@@ -1204,7 +1204,7 @@ def reroll(raw_idx, text_embeddings, class_order):
|
|
1204 |
min(len(babelnet_images[class_idx]), max_image_choices)
|
1205 |
)
|
1206 |
img_url = babelnet_images[class_idx][img_idx]["url"]
|
1207 |
-
class_labels =
|
1208 |
|
1209 |
if not precomputed_results:
|
1210 |
try:
|
@@ -1283,8 +1283,8 @@ with gr.Blocks(title="Babel-ImageNet Quiz") as demo:
|
|
1283 |
# Title Area
|
1284 |
gr.Markdown(
|
1285 |
"""
|
1286 |
-
#
|
1287 |
-
<small>
|
1288 |
"""
|
1289 |
)
|
1290 |
|
@@ -1313,7 +1313,7 @@ with gr.Blocks(title="Babel-ImageNet Quiz") as demo:
|
|
1313 |
|
1314 |
with gr.Column(scale=1):
|
1315 |
options = gr.Radio(
|
1316 |
-
choices=["Click", "start"
|
1317 |
interactive=False,
|
1318 |
label="Please click start to begin.",
|
1319 |
)
|
|
|
1029 |
no_image_idxs = [i for i, imgs in enumerate(babelnet_images) if len(imgs) == 0]
|
1030 |
IMG_HEIGHT, IMG_WIDTH = 512, 512
|
1031 |
|
1032 |
+
# precomputed_results = None
|
1033 |
+
# if os.path.exists("data/precomputed_results.json"):
|
1034 |
+
precomputed_results = json.load(open("data/precomputed_results.json"))
|
1035 |
|
1036 |
request_header = {
|
1037 |
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36"
|
|
|
1129 |
min(len(babelnet_images[class_idx]), max_image_choices)
|
1130 |
)
|
1131 |
img_url = babelnet_images[class_idx][img_idx]["url"]
|
1132 |
+
class_labels = openai_en_classes
|
1133 |
|
1134 |
if not precomputed_results:
|
1135 |
try:
|
|
|
1204 |
min(len(babelnet_images[class_idx]), max_image_choices)
|
1205 |
)
|
1206 |
img_url = babelnet_images[class_idx][img_idx]["url"]
|
1207 |
+
class_labels = openai_en_classes
|
1208 |
|
1209 |
if not precomputed_results:
|
1210 |
try:
|
|
|
1283 |
# Title Area
|
1284 |
gr.Markdown(
|
1285 |
"""
|
1286 |
+
# ¿Eres más lista 🤓 que CLIP 🤖?
|
1287 |
+
<small>Adapted from the original code by Gregor Geigle, WüNLP & Computer Vision Lab, University of Würzburg</small>
|
1288 |
"""
|
1289 |
)
|
1290 |
|
|
|
1313 |
|
1314 |
with gr.Column(scale=1):
|
1315 |
options = gr.Radio(
|
1316 |
+
choices=["Click", "start"],
|
1317 |
interactive=False,
|
1318 |
label="Please click start to begin.",
|
1319 |
)
|