Update app.py

app.py

@@ -110,13 +110,18 @@ if detect_button:
         image = Image.open(image)
         # Display the original image
         st.image(image, use_column_width=True, caption="Original Image")
+        
         # Perform object detection
         processed_image, detected_objects = perform_object_detection(image, detect_model, threshold)
-        # Display the detected objects
-        st.write(detected_objects)
+        
         # Display the image with detected objects
-        st.write(processed_image)  # testing to be deleted
-        st.image(processed_image, use_column_width=True, caption="Image with Detected Objects")
+        if isinstance(processed_image, Image.Image):
+            st.image(processed_image, use_column_width=True, caption="Image with Detected Objects")
+        else:
+            st.error("Failed to process image for object detection.")
 
+        # Display the detected objects as text
+        st.write(detected_objects)
     else:
         st.write("Please upload an image for object detection.")
+