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 |
)
|