Update handler.py
Browse files- handler.py +3 -2
handler.py
CHANGED
|
@@ -23,8 +23,9 @@ class EndpointHandler:
|
|
| 23 |
"""
|
| 24 |
from transformers import AutoProcessor, AutoModel
|
| 25 |
|
| 26 |
-
#
|
| 27 |
-
|
|
|
|
| 28 |
|
| 29 |
# Determine device
|
| 30 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 23 |
"""
|
| 24 |
from transformers import AutoProcessor, AutoModel
|
| 25 |
|
| 26 |
+
# Always load from the official Google model on HuggingFace Hub
|
| 27 |
+
# (path points to /repository which is our custom handler, not the model weights)
|
| 28 |
+
model_id = "google/siglip2-so400m-patch14-384"
|
| 29 |
|
| 30 |
# Determine device
|
| 31 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|