kokuma commited on
Commit
b16b259
·
verified ·
1 Parent(s): 4392bb0

Prepare for choices chage

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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
- 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,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 = babel_imagenet["EN"][1] if "EN" != "EN" else openai_en_classes
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 = babel_imagenet["EN"][1] if "EN" != "EN" else openai_en_classes
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
- # Are you smarter🤓 than 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,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", "to", "begin"],
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
  )