Update my_model/object_detection.py

my_model/object_detection.py

@@ -103,14 +103,18 @@ class ObjectDetector:
                 # Open the image from a file path
                 with Image.open(image_input) as image:
                     return image.convert("RGB")
-            else:
-                # Assume image_input is a file-like object and process it directly
+            elif hasattr(image_input, 'read'):
+                # If image_input is a file-like object, open it as an image
                 return Image.open(image_input).convert("RGB")
+            else:
+                # If image_input is already a PIL Image, just convert it
+                return image_input.convert("RGB")
         except Exception as e:
             print(f"Error processing image: {e}")
             raise
 
 
+
     def detect_objects(self, image, threshold=0.4):
         """
         Detect objects in the given image using the loaded model.