cccornflake commited on
Commit
0902778
·
verified ·
1 Parent(s): f590cac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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...")