Upload app.py

app.py

@@ -40,7 +40,7 @@ if uploaded_file is not None:
     img = tf.expand_dims(img, axis=0)
     
     pix2pix = get_model()
-    st.write("Model Loaded!!! Processing the image...")
-    pred = array_to_img(pix2pix.predict(img)[0] * 0.5 + 0.5)
-    st.image(pred)
-    _ = os.system("rm uploaded_image.png")
+    with st.spinner('Processing the image...'):
+        pred = array_to_img(pix2pix.predict(img)[0] * 0.5 + 0.5)
+        st.image(pred)
+        _ = os.system("rm uploaded_image.png")