yonatanbitton commited on
Commit
abf8ce5
Β·
1 Parent(s): c97e71d
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -35,6 +35,7 @@ emoji_to_label = {IMAGE: '🎨, πŸ§‘β€πŸŽ¨, πŸ’»', ANSWER: 'πŸ’‘, πŸ€”, πŸ§‘β€
35
  # batch_size = 16
36
  batch_size = 8
37
  target_size = (1024, 1024)
 
38
 
39
 
40
  def func(index):
@@ -98,18 +99,26 @@ def get_col(example):
98
 
99
 
100
  with demo:
 
101
  gr.Markdown("# Slide to iterate Visual Riddles")
102
 
103
  with gr.Column():
 
104
  num_batches = math.ceil(dataset_size / batch_size)
 
105
  slider = gr.Slider(minimum=0, maximum=num_batches, step=1, label=f'Page (out of {num_batches})')
 
106
  with gr.Row():
 
107
  index = slider.value
108
  start_index = 0 * batch_size
109
  end_index = start_index + batch_size
 
110
  all_examples = [Visual_Riddles[index] for index in list(range(start_index, end_index))]
 
111
  all_inputs_left_right = []
112
  for example_idx, example in enumerate(all_examples):
 
113
  inputs_left, text_inputs_right = get_col(example)
114
  inputs_left_right = inputs_left + text_inputs_right
115
  all_inputs_left_right += inputs_left_right
 
35
  # batch_size = 16
36
  batch_size = 8
37
  target_size = (1024, 1024)
38
+ print("HI3")
39
 
40
 
41
  def func(index):
 
99
 
100
 
101
  with demo:
102
+ print("HI4")
103
  gr.Markdown("# Slide to iterate Visual Riddles")
104
 
105
  with gr.Column():
106
+ print("HI5")
107
  num_batches = math.ceil(dataset_size / batch_size)
108
+ print("HI6")
109
  slider = gr.Slider(minimum=0, maximum=num_batches, step=1, label=f'Page (out of {num_batches})')
110
+ print("HI7")
111
  with gr.Row():
112
+ print("HI8")
113
  index = slider.value
114
  start_index = 0 * batch_size
115
  end_index = start_index + batch_size
116
+ print("HI9")
117
  all_examples = [Visual_Riddles[index] for index in list(range(start_index, end_index))]
118
+ print("HI10")
119
  all_inputs_left_right = []
120
  for example_idx, example in enumerate(all_examples):
121
+ print("HI11")
122
  inputs_left, text_inputs_right = get_col(example)
123
  inputs_left_right = inputs_left + text_inputs_right
124
  all_inputs_left_right += inputs_left_right