prakrutpatel commited on
Commit
b01c399
·
1 Parent(s): 6226ce3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -19,6 +19,8 @@ from IPython.display import display
19
  import io
20
  import pathlib
21
  print(tf.config.list_physical_devices('GPU'))
 
 
22
 
23
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
24
  if "models" in pathlib.Path.cwd().parts:
@@ -45,11 +47,11 @@ def load_model(model_dir):
45
  return model
46
  os.system('mkdir "Tortoise"')
47
  os.chdir('Tortoise/')
48
- os.system('curl -L "https://app.roboflow.com/ds/jCjxJgk04M?key=3JE38XqESy" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip')
49
  os.chdir('..')
50
  os.system('mkdir "COCO"')
51
  os.chdir('COCO/')
52
- os.system('curl -L "https://app.roboflow.com/ds/Yb2OGQm2xb?key=pbWEWpS5ec" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip')
53
  os.chdir('..')
54
 
55
  PATH_TO_TEST_IMAGES_DIR = pathlib.Path("COCO" + '/test/')
 
19
  import io
20
  import pathlib
21
  print(tf.config.list_physical_devices('GPU'))
22
+ RoboflowAPI = os.environ['RoboflowAPI']
23
+ RoboflowCocoAPI = os.environ['RoboflowCocoAPI']
24
 
25
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
26
  if "models" in pathlib.Path.cwd().parts:
 
47
  return model
48
  os.system('mkdir "Tortoise"')
49
  os.chdir('Tortoise/')
50
+ os.system(f'curl -L "https://app.roboflow.com/ds/{RoboflowAPI}" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip')
51
  os.chdir('..')
52
  os.system('mkdir "COCO"')
53
  os.chdir('COCO/')
54
+ os.system(f'curl -L "https://app.roboflow.com/ds/{RoboflowCocoAPI}" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip')
55
  os.chdir('..')
56
 
57
  PATH_TO_TEST_IMAGES_DIR = pathlib.Path("COCO" + '/test/')