Update handler.py

handler.py

@@ -7,8 +7,8 @@ import logging
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.processor = AutoProcessor.from_pretrained("Blueway/inference-endpoint-zero-shot-image-classification")
-        self.model = OmDetTurboForObjectDetection.from_pretrained("Blueway/inference-endpoint-zero-shot-image-classification")
+        self.processor = AutoProcessor.from_pretrained("Blueway/inference-endpoint-for-omdet-turbo-swin-tiny-hf")
+        self.model = OmDetTurboForObjectDetection.from_pretrained("Blueway/inference-endpoint-for-omdet-turbo-swin-tiny-hf")
        
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
@@ -33,8 +33,6 @@ class EndpointHandler():
             score_threshold=0.3,
             nms_threshold=0.3,
         )[0]
-        print(results)
-        logging.error(results)
         # Convert tensors to lists
         serializable_results = {
             'boxes': results['boxes'].tolist(),