added depth_map

handler.py

@@ -12,6 +12,7 @@ class EndpointHandler:
     def __init__(self, path=""):
         # Load model and processor
         self.model_path = path or os.environ.get("MODEL_PATH", "")
+        print(self.model_path)
         self.image_processor = AutoImageProcessor.from_pretrained(self.model_path)
         self.model = AutoModelForDepthEstimation.from_pretrained(self.model_path)
 

test_hf.py

@@ -6,8 +6,8 @@ from PIL import Image
 import io
 
 # URL del endpoint proporcionado por Hugging Face
-# ENDPOINT_URL = "https://qh7glc3xj9iw4tk2.eu-west-1.aws.endpoints.huggingface.cloud"
-ENDPOINT_URL = os.environ.get("ENDPOINT_URL", "")
+ENDPOINT_URL = "https://xeagnn3ecuk0ol15.us-east-1.aws.endpoints.huggingface.cloud"
+# ENDPOINT_URL = os.environ.get("ENDPOINT_URL", "")
 # Token de API de Hugging Face
 # API_TOKEN = "hf_..."
 API_TOKEN = os.environ.get("API_TOKEN", "")
@@ -17,28 +17,28 @@ headers = {
 }
 
 # Cargar y codificar una imagen
-image = Image.open("mine.jpeg")
-buffered = io.BytesIO()
-image.save(buffered, format="JPEG")
-img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+# image = Image.open("mine.jpeg")
+# buffered = io.BytesIO()
+# image.save(buffered, format="JPEG")
+# img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
 # Preparar los datos para la solicitud
+payload = {
+    "inputs"       : {
+
+    },
+    "url"          : "https://images.unsplash.com/photo-1586023492125-27b2c045efd7?fm=jpg&q=60&w=3000&ixlib=rb-4.0.3&ixid=M3wxMjA3fDB8MHxzZWFyY2h8Mnx8aW50ZXJpb3IlMjBkZXNpZ258ZW58MHx8MHx8fDA%3D",
+    "visualization": False
+}
+
 # payload = {
 #     "inputs"       : {
 #
 #     },
-#     "url"          : "https://images.unsplash.com/photo-1586023492125-27b2c045efd7?fm=jpg&q=60&w=3000&ixlib=rb-4.0.3&ixid=M3wxMjA3fDB8MHxzZWFyY2h8Mnx8aW50ZXJpb3IlMjBkZXNpZ258ZW58MHx8MHx8fDA%3D",
+#     "file"         : img_str,
 #     "visualization": True
 # }
 
-payload = {
-    "inputs"       : {
-
-    },
-    "file"         : img_str,
-    "visualization": True
-}
-
 # Enviar la solicitud
 response = requests.post(ENDPOINT_URL, headers=headers, json=payload)