yonatanbitton commited on
Commit
230aa4d
Β·
1 Parent(s): 4e4df51

adding explorer

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -27,13 +27,6 @@ emoji_to_label = {IMAGE_DESIGNER: '🎨, πŸ§‘β€πŸŽ¨, πŸ’»', DESIGNER_EXPLANATIO
27
  QA: '❓, πŸ€”, πŸ’‘', IMAGE_ID: 'πŸ”, πŸ“„, πŸ’Ύ', COMMONSENSE_CATEGORY: 'πŸ€”, πŸ“š, πŸ’‘', SELECTED_CAPTION: 'πŸ“, πŸ‘Œ, πŸ’¬'}
28
  target_size = (1024, 1024)
29
 
30
-
31
- def func(index):
32
- example = whoops[index]
33
- values = get_instance_values(example)
34
- return values
35
-
36
-
37
  def get_instance_values(example):
38
  values = []
39
  for k in left_side_columns + right_side_columns:
@@ -52,15 +45,15 @@ def list_to_string(lst):
52
  return '\n'.join(['{}. {}'.format(i + 1, item) for i, item in enumerate(lst)])
53
 
54
  def plot_image(index):
55
- example = whoops[index]
56
  instance_values = get_instance_values(example)
57
  assert len(left_side_columns) == len(
58
  instance_values[:len(left_side_columns)]) # excluding the image & designer
59
  for key, value in zip(left_side_columns, instance_values[:len(left_side_columns)]):
60
  if key == IMAGE:
61
- img = whoops[index]["image"]
62
  img_resized = img.resize(target_size)
63
- gr.Image(value=img_resized, label=whoops[index]['commonsense_category'])
64
  else:
65
  label = key.capitalize().replace("_", " ")
66
  gr.Textbox(value=value, label=f"{label} {emoji_to_label[key]}")
@@ -74,7 +67,7 @@ def plot_image(index):
74
 
75
  columns_number = 4
76
  # rows_number = int(dataset_size / columns_number)
77
- rows_number = 30
78
  whoops_sample = whoops.shuffle().select(range(0, columns_number * rows_number))
79
  index = 0
80
 
 
27
  QA: '❓, πŸ€”, πŸ’‘', IMAGE_ID: 'πŸ”, πŸ“„, πŸ’Ύ', COMMONSENSE_CATEGORY: 'πŸ€”, πŸ“š, πŸ’‘', SELECTED_CAPTION: 'πŸ“, πŸ‘Œ, πŸ’¬'}
28
  target_size = (1024, 1024)
29
 
 
 
 
 
 
 
 
30
  def get_instance_values(example):
31
  values = []
32
  for k in left_side_columns + right_side_columns:
 
45
  return '\n'.join(['{}. {}'.format(i + 1, item) for i, item in enumerate(lst)])
46
 
47
  def plot_image(index):
48
+ example = whoops_sample[index]
49
  instance_values = get_instance_values(example)
50
  assert len(left_side_columns) == len(
51
  instance_values[:len(left_side_columns)]) # excluding the image & designer
52
  for key, value in zip(left_side_columns, instance_values[:len(left_side_columns)]):
53
  if key == IMAGE:
54
+ img = whoops_sample[index]["image"]
55
  img_resized = img.resize(target_size)
56
+ gr.Image(value=img_resized, label=whoops_sample[index]['commonsense_category'])
57
  else:
58
  label = key.capitalize().replace("_", " ")
59
  gr.Textbox(value=value, label=f"{label} {emoji_to_label[key]}")
 
67
 
68
  columns_number = 4
69
  # rows_number = int(dataset_size / columns_number)
70
+ rows_number = 25
71
  whoops_sample = whoops.shuffle().select(range(0, columns_number * rows_number))
72
  index = 0
73