Update app.py

app.py

@@ -24,6 +24,7 @@ st.sidebar.markdown('<p class="font">Afrodreams.AI</p>', unsafe_allow_html=True)
 with st.sidebar.expander("About the App"):
      st.write("""
         This app takes in your image and styles it with a unique african art.""")
+     style_weight = st.slider("Select Style Weight)
 
 #Add file uploader to allow users to upload photos
 uploaded_file = st.file_uploader("", type=['jpg','png','jpeg'])
@@ -42,29 +43,32 @@ if uploaded_file is not None:
         
     # add a button
     run = st.button('Generate Art')
+    my_bar = st.progress(0)
+    params = neural_style.TransferParams()
+    params.gpu = "c"
+    params.backend = "mkl"
+    params.image_size = 300
+    params.content_image = uploaded_file
+    path = 'stylesv2'
+    styles = os.listdir(path)
+    params.style_weight = 12
+    params.style_image = path + '/' + random.choice(styles)
+
     if run==True:
-        
-        with st.spinner('Wait for it...'):
-            params = neural_style.TransferParams()
-            params.gpu = "c"
-            params.backend = "mkl"
-            params.image_size = 300
-            params.content_image = uploaded_file
-            path = 'stylesv2'
-            styles = os.listdir(path)
-            params.style_image = path + '/' + random.choice(styles)
-            neural_style.transfer(params)
+        st.session_state.submitted = True    
+        my_bar.progress(neural_style.transfer(params))
 
         #display image when done.
         with col2:
-            result = Image.open('out.png')
-            st.image(result, width=500)
-            buf = BytesIO()
-            result.save(buf, format="JPEG")
-            byte_im = buf.getvalue()
-            run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
-
-
+            if 'submitted' in st.session_state:
+                result = Image.open('out.png')
+                st.image(result, width=500)
+                buf = BytesIO()
+                result.save(buf, format="JPEG")
+                byte_im = buf.getvalue()
+                run =st.download_button(label="Download Image", data=byte_im, file_name='afrodreams.jpg', mime="image/png")
+    
+