Update app.py

app.py

@@ -114,20 +114,21 @@ 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)
+            
+            if processed_image:
+                # Display the image with detected objects
+                st.image(processed_image, use_column_width=True, caption="Image with Detected Objects")
+                # Display the detected objects as text
+                st.write(detected_objects)
+            else:
+                st.error("Failed to process image for object detection.")
+
         except Exception as e:
             st.error(f"Error loading image: {e}")
-            return
 
-        # Perform object detection
-        processed_image, detected_objects = perform_object_detection(image, detect_model, threshold)
-        
-        if processed_image:
-            # Display the image with detected objects
-            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.")
+