JunhanCai commited on
Commit
37570e3
·
verified ·
1 Parent(s): 2dbee01

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. webserver/app.py +1 -2
webserver/app.py CHANGED
@@ -36,8 +36,7 @@ UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
36
  RUNS_DIR = os.path.join(BASE_DIR, "runs")
37
  PREDICTIONS_DIR = os.path.join(BASE_DIR, "predictions")
38
  TEMPLATE_DIR = os.path.join(BASE_DIR, "templates")
39
- ROOT_DIR = os.path.dirname(BASE_DIR)
40
- DEFAULT_MODEL_PATH = os.path.join(ROOT_DIR, "weights", "Fine_tuned.pth")
41
  os.makedirs(UPLOAD_DIR, exist_ok=True)
42
  os.makedirs(RUNS_DIR, exist_ok=True)
43
  os.makedirs(PREDICTIONS_DIR, exist_ok=True)
 
36
  RUNS_DIR = os.path.join(BASE_DIR, "runs")
37
  PREDICTIONS_DIR = os.path.join(BASE_DIR, "predictions")
38
  TEMPLATE_DIR = os.path.join(BASE_DIR, "templates")
39
+ DEFAULT_MODEL_PATH = os.path.join(BASE_DIR, "weights", "Fine_tuned.pth")
 
40
  os.makedirs(UPLOAD_DIR, exist_ok=True)
41
  os.makedirs(RUNS_DIR, exist_ok=True)
42
  os.makedirs(PREDICTIONS_DIR, exist_ok=True)