yonatanbitton commited on
Commit
c97e71d
·
1 Parent(s): 7044f7a
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -25,7 +25,7 @@ CATEGORY = "Category"
25
  DESIGNER = "Designer"
26
 
27
 
28
-
29
  left_side_columns = [IMAGE]
30
  right_side_columns = [x for x in Visual_Riddles.features.keys() if x not in left_side_columns]
31
  # enumerate_cols = [CROWD_CAPTIONS, CROWD_EXPLANATIONS, CROWD_UNDERSPECIFIED_CAPTIONS]
@@ -38,6 +38,7 @@ target_size = (1024, 1024)
38
 
39
 
40
  def func(index):
 
41
  start_index = index * batch_size
42
  end_index = start_index + batch_size
43
  all_examples = [Visual_Riddles[index] for index in list(range(start_index, end_index))]
@@ -49,15 +50,16 @@ def func(index):
49
 
50
 
51
  def get_instance_values(example):
 
52
  values = []
53
  for k in left_side_columns + right_side_columns:
54
  if k == IMAGE:
55
  value = example["Image"].resize(target_size)
56
- elif k in enumerate_cols:
57
- value = list_to_string(example[k])
58
- elif k == QA:
59
- qa_list = [f"Q: {x[0]} A: {x[1]}" for x in example[k]]
60
- value = list_to_string(qa_list)
61
  else:
62
  value = example[k]
63
  values.append(value)
@@ -69,6 +71,7 @@ demo = gr.Blocks()
69
 
70
 
71
  def get_col(example):
 
72
  instance_values = get_instance_values(example)
73
  with gr.Column():
74
  inputs_left = []
 
25
  DESIGNER = "Designer"
26
 
27
 
28
+ print("HI2")
29
  left_side_columns = [IMAGE]
30
  right_side_columns = [x for x in Visual_Riddles.features.keys() if x not in left_side_columns]
31
  # enumerate_cols = [CROWD_CAPTIONS, CROWD_EXPLANATIONS, CROWD_UNDERSPECIFIED_CAPTIONS]
 
38
 
39
 
40
  def func(index):
41
+ print("func")
42
  start_index = index * batch_size
43
  end_index = start_index + batch_size
44
  all_examples = [Visual_Riddles[index] for index in list(range(start_index, end_index))]
 
50
 
51
 
52
  def get_instance_values(example):
53
+ print("get_instance_values")
54
  values = []
55
  for k in left_side_columns + right_side_columns:
56
  if k == IMAGE:
57
  value = example["Image"].resize(target_size)
58
+ # elif k in enumerate_cols:
59
+ # value = list_to_string(example[k])
60
+ # elif k == QA:
61
+ # qa_list = [f"Q: {x[0]} A: {x[1]}" for x in example[k]]
62
+ # value = list_to_string(qa_list)
63
  else:
64
  value = example[k]
65
  values.append(value)
 
71
 
72
 
73
  def get_col(example):
74
+ print("get_col")
75
  instance_values = get_instance_values(example)
76
  with gr.Column():
77
  inputs_left = []