Debug - Delpoyment

.gitignore

@@ -1,2 +1,3 @@
 *env*
-*.DS_Store*
+*.DS_Store*
+*__pycache__*

requirements.txt

@@ -1,27 +1,5 @@
-accelerate==1.3.0
-certifi==2024.12.14
-charset-normalizer==3.4.1
-filelock==3.17.0
-fsspec==2024.12.0
-huggingface-hub==0.27.1
-idna==3.10
-Jinja2==3.1.5
-MarkupSafe==3.0.2
-mpmath==1.3.0
-networkx==3.4.2
-numpy==2.2.2
-packaging==24.2
-pillow==11.1.0
-psutil==6.1.1
-PyYAML==6.0.2
-regex==2024.11.6
-requests==2.32.3
-safetensors==0.5.2
-setuptools==75.8.0
-sympy==1.13.1
-tokenizers==0.21.0
 torch==2.5.1
-tqdm==4.67.1
 transformers==4.48.1
-typing_extensions==4.12.2
-urllib3==2.3.0
+accelerate==1.3.0
+pillow==11.1.0
+

test_handler.py

@@ -0,0 +1,29 @@
+import base64
+from io import BytesIO
+import requests
+from PIL import Image
+
+# Import the EndpointHandler class
+from handler import EndpointHandler
+
+# Fetch the image from the URL
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+response = requests.get(url, stream=True)
+image = Image.open(response.raw).convert("RGB")  # Ensure the image is in RGB format
+
+# Encode the image as base64
+buffered = BytesIO()
+image.save(buffered, format="JPEG")  # Save the image to a buffer in JPEG format
+encoded_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+# Prepare the input data
+data = {"images": [encoded_image]}
+
+# Initialize the handler
+handler = EndpointHandler()
+
+# Process the image
+result = handler(data)
+
+# Print the result
+print(result)