Update app.py

app.py

@@ -95,10 +95,15 @@ def perform_object_detection(image, model_name, threshold):
     Returns:
     PIL.Image, str: The image with drawn bounding boxes and a string of detected objects.
     """
-    
+
+    # Initialize the ObjectDetector
+    detector = ObjectDetector()
+    # Load the specified model
+    detector.load_model(model_name)
+
     try:
         # Perform object detection and draw bounding boxes
-        processed_image, detected_objects = detect_and_draw_objects(image, model_name, threshold)
+        processed_image, detected_objects = detector.detect_and_draw_objects(image, model_name, threshold)
         return processed_image, detected_objects
     except Exception as e:
         # Print the error for debugging
@@ -132,3 +137,4 @@ if detect_button:
     else:
         st.write("Please upload an image for object detection.")
 
+