hieupt commited on
Commit
637b1dc
·
1 Parent(s): 4effec7

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