GabrielML commited on
Commit
8717d3d
·
1 Parent(s): efde35f

remove debug

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -20,7 +20,6 @@ from tqdm import tqdm
20
  from util import transform
21
 
22
  IMAGE_PATH = os.path.join(os.getcwd(), 'src/examples')
23
- print(os.listdir(os.path.join(IMAGE_PATH, 'antelope')))
24
  RANDOM_IMAGES_TO_SHOW = 10
25
  IMAGES_PER_ROW = 5
26
 
@@ -44,7 +43,6 @@ def load_random_images():
44
  p = p.replace('\\', '/')
45
  p = p.replace('//', '/')
46
  animal = data_df.iloc[idx]['target']
47
- print(p)
48
  if os.path.exists(p):
49
  random_images.append((animal, Image.open(p)))
50
  return random_images
@@ -55,7 +53,7 @@ def get_class_name(idx):
55
  @lru_cache(maxsize=100)
56
  def get_translated(to_translate):
57
  return GoogleTranslator(source="en", target="de").translate(to_translate)
58
- for idx in tqdm(range(90), desc="Translate animals"): get_translated(get_class_name(idx))
59
 
60
  def infer_image(image):
61
  image = transform(image)
 
20
  from util import transform
21
 
22
  IMAGE_PATH = os.path.join(os.getcwd(), 'src/examples')
 
23
  RANDOM_IMAGES_TO_SHOW = 10
24
  IMAGES_PER_ROW = 5
25
 
 
43
  p = p.replace('\\', '/')
44
  p = p.replace('//', '/')
45
  animal = data_df.iloc[idx]['target']
 
46
  if os.path.exists(p):
47
  random_images.append((animal, Image.open(p)))
48
  return random_images
 
53
  @lru_cache(maxsize=100)
54
  def get_translated(to_translate):
55
  return GoogleTranslator(source="en", target="de").translate(to_translate)
56
+ for idx in range(90): get_translated(get_class_name(idx))
57
 
58
  def infer_image(image):
59
  image = transform(image)