update req

handler.py

@@ -27,6 +27,8 @@ class EndpointHandler():
         embedding = torch.mean(last_hidden_state, dim=1).flatten().tolist()
         return {"embedding": embedding}
 
+"""
 handler = EndpointHandler()
 output = handler({"inputs": "https://figma-staging-api.s3.us-west-2.amazonaws.com/images/a8c6a0cc-c022-4f3a-9fc5-ac8582c964dd"})
 print(output)
+"""

requirements.txt

@@ -1,22 +1,3 @@
-certifi==2023.7.22
-charset-normalizer==3.3.2
-filelock==3.13.1
-fsspec==2023.10.0
-huggingface-hub==0.17.3
-idna==3.4
-Jinja2==3.1.2
-MarkupSafe==2.1.3
-mpmath==1.3.0
-networkx==3.2.1
-numpy==1.26.2
-packaging==23.2
 Pillow==10.1.0
-PyYAML==6.0.1
-regex==2023.10.3
-requests==2.31.0
-safetensors==0.4.0
-sympy==1.12
-tokenizers==0.14.1
-tqdm==4.66.1
-typing_extensions==4.8.0
-urllib3==2.1.0
+torch==2.1.0
+transformers==4.35.0