Update app.py

app.py

@@ -58,8 +58,9 @@ if uploaded_file is not None:
             styles = os.listdir(path)
             params.style_image = path + '/' + random.choice(styles)
             
-        st.session_state.submitted = True    
-        my_bar.progress(neural_style.transfer(params))
+        st.session_state.submitted = True   
+        with st.spinner('Wait for it...'): 
+            neural_style.transfer(params)
 
         #display image when done.
         with col2: