Owos commited on
Commit
3713c83
·
1 Parent(s): 81ec8c5

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ if uploaded_file is not None:
49
  params.content_image = uploaded_file
50
  path = 'styles'
51
  styles = os.listdir(path)
52
- params.style_image = random.choice(styles)
53
  neural_style.transfer(params)
54
 
55
  #display image when done.
 
49
  params.content_image = uploaded_file
50
  path = 'styles'
51
  styles = os.listdir(path)
52
+ params.style_image = path + '/' + random.choice(styles)
53
  neural_style.transfer(params)
54
 
55
  #display image when done.