Spaces:
Sleeping
Sleeping
Commit
·
b01c399
1
Parent(s):
6226ce3
Update app.py
Browse files
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/
|
49 |
os.chdir('..')
|
50 |
os.system('mkdir "COCO"')
|
51 |
os.chdir('COCO/')
|
52 |
-
os.system('curl -L "https://app.roboflow.com/ds/
|
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/')
|