Update my_model/object_detection.py

my_model/object_detection.py

@@ -87,27 +87,30 @@ class ObjectDetector:
             print(f"Error loading YOLOv5 model: {e}")
             raise
 
-    def process_image(self, image_path):
+    def process_image(self, image_input):
         """
-        Process the image from the given path.
-
+        Process the image from the given path or file-like object.
         Args:
-            image_path (str): Path to the image file.
-
+            image_input (str or file-like object): Path to the image file or a file-like object.
         Returns:
             Image.Image: Processed image in RGB format.
-
         Raises:
             Exception: If an error occurs during image processing.
         """
-
         try:
-            with Image.open(image_path) as image:
-                return image.convert("RGB")
+            # Check if the input is a string (path) or a file-like object
+            if isinstance(image_input, str):
+                # 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
+                return Image.open(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.