Update model.py

Used pipeline to load model

model.py

@@ -1,5 +1,6 @@
 import albumentations as A
 from transformers import AutoModelForObjectDetection, AutoImageProcessor
+from transformers import pipeline
 
 
 # Mapping for labels and IDs
@@ -52,11 +53,15 @@ def get_val_transform():
     )
 
 # Load the model
-def load_model():
-    model_path = "sergiopaniego/detr-resnet-50-dc5-fashionpedia-finetuned"  # Path to your saved model
-    image_processor = AutoImageProcessor.from_pretrained(model_path)
-    model = AutoModelForObjectDetection.from_pretrained(
-        model_path,  # This will automatically use model.safetensors
-        ignore_mismatched_sizes=True
+def load_model(device: int = -1):
+    """
+    Load the DETR model pipeline.
+    :param device: Specify device to load the model (-1 for CPU, 0 for GPU).
+    :return: Hugging Face object-detection pipeline.
+    """
+    model_pipeline = pipeline(
+        "object-detection", 
+        model="sergiopaniego/detr-resnet-50-dc5-fashionpedia-finetuned", 
+        device=device
     )
-    return model, image_processor
+    return model_pipeline