Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import cv2
 import numpy as np
 from PIL import Image
+from io import BytesIO
 
 st.set_page_config(page_title="Image to Sketch", layout="centered")
 st.title("🖼️ Image to Sketch Converter")
@@ -27,3 +28,19 @@ if uploaded_file is not None:
     st.subheader("Sketch Output")
     st.image(sketch, use_container_width=True, channels="GRAY")
 
+    # Convert sketch to PIL image
+    sketch_pil = Image.fromarray(sketch)
+
+    # Save image to memory
+    buf = BytesIO()
+    sketch_pil.save(buf, format="PNG")
+    byte_im = buf.getvalue()
+
+    # Download button
+    st.download_button(
+        label="📥 Download Sketch",
+        data=byte_im,
+        file_name="sketch.png",
+        mime="image/png"
+    )
+