hieupt commited on
Commit
0a1d473
·
1 Parent(s): d989b7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,12 +10,12 @@ from PIL import Image
10
  from pathlib import Path
11
 
12
  # Get model
13
- model_dir = 'models'
14
  models = list(Path(model_dir).glob("*/*.pth.tar"))
15
  models = sorted(models)
16
 
17
  # Get style image
18
- style_dir = 'style_images'
19
  style_list = list(Path(style_dir).glob("*"))
20
  style_list = sorted(style_list)
21
 
 
10
  from pathlib import Path
11
 
12
  # Get model
13
+ model_dir = 'models/'
14
  models = list(Path(model_dir).glob("*/*.pth.tar"))
15
  models = sorted(models)
16
 
17
  # Get style image
18
+ style_dir = 'style_images/'
19
  style_list = list(Path(style_dir).glob("*"))
20
  style_list = sorted(style_list)
21