leonelhs commited on
Commit
829edc1
1 Parent(s): cd7fe04

fixing dirs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ device.set(device=DeviceId.CPU)
16
  REPO_ID = "leonelhs/deoldify"
17
  MODEL_NAME = "ColorizeArtistic_gen.pth"
18
 
19
- model_path = Path("models")
20
  snapshot_folder = snapshot_download(repo_id=REPO_ID)
21
  model_path_hf = Path(snapshot_folder).joinpath(MODEL_NAME)
22
 
 
16
  REPO_ID = "leonelhs/deoldify"
17
  MODEL_NAME = "ColorizeArtistic_gen.pth"
18
 
19
+ model_path = Path("./models")
20
  snapshot_folder = snapshot_download(repo_id=REPO_ID)
21
  model_path_hf = Path(snapshot_folder).joinpath(MODEL_NAME)
22