Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,8 +36,6 @@ if uploaded_file is not None:
|
|
| 36 |
# Read and display original image
|
| 37 |
img = Image.open(uploaded_file)
|
| 38 |
img = np.array(img)
|
| 39 |
-
|
| 40 |
-
st.image(img, caption='Original Image', use_column_width=True)
|
| 41 |
|
| 42 |
# Sliders for parameters
|
| 43 |
median_blur_value = st.slider("Median Blur Value", min_value=3, max_value=11, value=5, step=2)
|
|
@@ -47,11 +45,17 @@ if uploaded_file is not None:
|
|
| 47 |
sigma_color = st.slider("Sigma Color", min_value=0, max_value=400, value=300)
|
| 48 |
sigma_space = st.slider("Sigma Space", min_value=0, max_value=400, value=300)
|
| 49 |
|
| 50 |
-
# Cartoonize the image with selected parameters
|
| 51 |
cartoon_image = cartoonize_image(img, median_blur_value, block_size, c_value, d_value, sigma_color, sigma_space)
|
| 52 |
|
| 53 |
-
# Display cartoonized
|
| 54 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
|
| 56 |
# Download button for cartoonized image
|
| 57 |
if st.button("Download Cartoonized Image"):
|
|
|
|
| 36 |
# Read and display original image
|
| 37 |
img = Image.open(uploaded_file)
|
| 38 |
img = np.array(img)
|
|
|
|
|
|
|
| 39 |
|
| 40 |
# Sliders for parameters
|
| 41 |
median_blur_value = st.slider("Median Blur Value", min_value=3, max_value=11, value=5, step=2)
|
|
|
|
| 45 |
sigma_color = st.slider("Sigma Color", min_value=0, max_value=400, value=300)
|
| 46 |
sigma_space = st.slider("Sigma Space", min_value=0, max_value=400, value=300)
|
| 47 |
|
| 48 |
+
# Cartoonize the image with selected parameters in real-time
|
| 49 |
cartoon_image = cartoonize_image(img, median_blur_value, block_size, c_value, d_value, sigma_color, sigma_space)
|
| 50 |
|
| 51 |
+
# Display original and cartoonized images side by side
|
| 52 |
+
col1, col2 = st.columns(2)
|
| 53 |
+
|
| 54 |
+
with col1:
|
| 55 |
+
st.image(img, caption='Original Image', use_column_width=True)
|
| 56 |
+
|
| 57 |
+
with col2:
|
| 58 |
+
st.image(cartoon_image, caption='Cartoonized Image', use_column_width=True)
|
| 59 |
|
| 60 |
# Download button for cartoonized image
|
| 61 |
if st.button("Download Cartoonized Image"):
|