Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,9 +6,12 @@ import os
|
|
6 |
|
7 |
REPO_ID = 'vaivTA/yolov8x_doclaynet'
|
8 |
FILENAME = "weights/best.pt"
|
9 |
-
|
|
|
|
|
10 |
if not os.path.exists('/home/user/app/weights/best.pt') :
|
11 |
|
|
|
12 |
os.makedirs('/home/user/app/weights', exist_ok=True)
|
13 |
os.chdir('/home/user/app')
|
14 |
print("downloading model...")
|
|
|
6 |
|
7 |
REPO_ID = 'vaivTA/yolov8x_doclaynet'
|
8 |
FILENAME = "weights/best.pt"
|
9 |
+
|
10 |
+
if os.path.exists('/home/user/app/weights/best.pt'):
|
11 |
+
print("model exist.")
|
12 |
if not os.path.exists('/home/user/app/weights/best.pt') :
|
13 |
|
14 |
+
|
15 |
os.makedirs('/home/user/app/weights', exist_ok=True)
|
16 |
os.chdir('/home/user/app')
|
17 |
print("downloading model...")
|