Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -46,8 +46,8 @@ st.markdown("""
|
|
46 |
Upload an image, and get the sketch version instantly!
|
47 |
""")
|
48 |
|
49 |
-
# Example conversions
|
50 |
-
|
51 |
|
52 |
# Replace 'Dog.jpg' with the path to an example image if running locally
|
53 |
example_image_path = 'Dog.jpg'
|
@@ -58,40 +58,46 @@ if example_image is not None:
|
|
58 |
example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
|
59 |
example_sketch = image_to_sketch(example_image)
|
60 |
|
|
|
|
|
61 |
with col1:
|
62 |
-
st.subheader("Example Conversions")
|
63 |
st.image(example_image_rgb, caption='Original Image', use_column_width=True)
|
|
|
64 |
st.image(example_sketch, caption='Sketch Image', use_column_width=True)
|
65 |
else:
|
66 |
st.error(f"Failed to load example image from path: {example_image_path}")
|
67 |
|
68 |
-
|
69 |
-
|
70 |
-
|
71 |
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
|
|
|
|
|
|
78 |
|
79 |
-
|
80 |
-
|
|
|
81 |
st.image(sketch, caption='Sketch', use_column_width=True)
|
82 |
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
-
|
95 |
|
96 |
# Footer
|
97 |
st.markdown("""
|
|
|
46 |
Upload an image, and get the sketch version instantly!
|
47 |
""")
|
48 |
|
49 |
+
# Example conversions
|
50 |
+
st.subheader("Example Conversions")
|
51 |
|
52 |
# Replace 'Dog.jpg' with the path to an example image if running locally
|
53 |
example_image_path = 'Dog.jpg'
|
|
|
58 |
example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
|
59 |
example_sketch = image_to_sketch(example_image)
|
60 |
|
61 |
+
col1, col2 = st.columns(2)
|
62 |
+
|
63 |
with col1:
|
|
|
64 |
st.image(example_image_rgb, caption='Original Image', use_column_width=True)
|
65 |
+
with col2:
|
66 |
st.image(example_sketch, caption='Sketch Image', use_column_width=True)
|
67 |
else:
|
68 |
st.error(f"Failed to load example image from path: {example_image_path}")
|
69 |
|
70 |
+
# User upload section
|
71 |
+
st.subheader("Upload Your Image")
|
72 |
+
uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
|
73 |
|
74 |
+
if uploaded_file is not None:
|
75 |
+
image = np.array(Image.open(uploaded_file))
|
76 |
+
st.image(image, caption='Uploaded Image', use_column_width=True)
|
77 |
+
|
78 |
+
st.write("Converting...")
|
79 |
+
|
80 |
+
sketch = image_to_sketch(image)
|
81 |
+
|
82 |
+
col3, col4 = st.columns(2)
|
83 |
|
84 |
+
with col3:
|
85 |
+
st.image(image, caption='Uploaded Image', use_column_width=True)
|
86 |
+
with col4:
|
87 |
st.image(sketch, caption='Sketch', use_column_width=True)
|
88 |
|
89 |
+
# Provide a download link for the sketch image
|
90 |
+
im_pil = Image.fromarray(sketch)
|
91 |
+
im_pil.save("sketch.png")
|
92 |
+
with open("sketch.png", "rb") as file:
|
93 |
+
btn = st.download_button(
|
94 |
+
label="Download Sketch",
|
95 |
+
data=file,
|
96 |
+
file_name="sketch.png",
|
97 |
+
mime="image/png"
|
98 |
+
)
|
99 |
+
else:
|
100 |
+
st.info("Please upload an image to convert.")
|
101 |
|
102 |
# Footer
|
103 |
st.markdown("""
|