pvyas96 commited on
Commit
1146b1f
·
verified ·
1 Parent(s): ace25d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
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 image
54
- st.image(cartoon_image, caption='Cartoonized Image', use_column_width=True)
 
 
 
 
 
 
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"):