Update app.py

app.py

@@ -73,7 +73,6 @@ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png
 
 if uploaded_file is not None:
     image = np.array(Image.open(uploaded_file))
-    st.image(image, caption='Uploaded Image', use_column_width=True)
     
     st.write("Converting...")
 
@@ -87,15 +86,13 @@ if uploaded_file is not None:
         st.image(sketch, caption='Sketch', use_column_width=True)
 
     # Provide a download link for the sketch image
-    im_pil = Image.fromarray(sketch)
-    im_pil.save("sketch.png")
-    with open("sketch.png", "rb") as file:
-        btn = st.download_button(
-            label="Download Sketch",
-            data=file,
-            file_name="sketch.png",
-            mime="image/png"
-        )
+    st.markdown("<br>", unsafe_allow_html=True)  # Add some space before the button
+    btn = st.download_button(
+        label="Download Sketch",
+        data=Image.fromarray(sketch).tobytes(),
+        file_name="sketch.png",
+        mime="image/png"
+    )
 else:
     st.info("Please upload an image to convert.")