Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.
|
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.
|
|
|
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 |
|