Owos commited on
Commit
ac152f1
·
1 Parent(s): 8379262

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -49,14 +49,15 @@ if uploaded_file is not None:
49
  params.backend = "mkl"
50
  params.image_size = 300
51
  params.content_image = uploaded_file
52
- params.style_image = path + '/' + random.choice(styles)
53
  keep_style = False
54
  if run==True:
55
  # run image selection if keep style is false
56
  if keep_style==False:
57
  path = 'stylesv2'
58
  styles = os.listdir(path)
59
- params.style_weight = style_weight
 
60
  st.session_state.submitted = True
61
  my_bar.progress(neural_style.transfer(params))
62
 
 
49
  params.backend = "mkl"
50
  params.image_size = 300
51
  params.content_image = uploaded_file
52
+ params.style_weight = style_weight
53
  keep_style = False
54
  if run==True:
55
  # run image selection if keep style is false
56
  if keep_style==False:
57
  path = 'stylesv2'
58
  styles = os.listdir(path)
59
+ params.style_image = path + '/' + random.choice(styles)
60
+
61
  st.session_state.submitted = True
62
  my_bar.progress(neural_style.transfer(params))
63