Update app.py

app.py

@@ -50,13 +50,20 @@ st.markdown("""
 col1, col2 = st.columns(2)
 
 # Replace 'Dog.jpg' with the path to an example image if running locally
-example_image = cv2.imread('Dog.jpg')
-example_sketch = image_to_sketch(example_image)
+example_image_path = 'Dog.jpg'
+example_image = cv2.imread(example_image_path)
 
-with col1:
-    st.subheader("Example Conversions")
-    st.image(example_image, caption='Original Image', use_column_width=True)
-    st.image(example_sketch, caption='Sketch Image', use_column_width=True)
+if example_image is not None:
+    # Convert BGR to RGB for correct color display
+    example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
+    example_sketch = image_to_sketch(example_image)
+
+    with col1:
+        st.subheader("Example Conversions")
+        st.image(example_image_rgb, caption='Original Image', use_column_width=True)
+        st.image(example_sketch, caption='Sketch Image', use_column_width=True)
+else:
+    st.error(f"Failed to load example image from path: {example_image_path}")
 
 with col2:
     st.subheader("Upload Your Image")