peterproofpath commited on
Commit
32936bc
·
verified ·
1 Parent(s): c71b705

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -23,8 +23,9 @@ class EndpointHandler:
23
  """
24
  from transformers import AutoProcessor, AutoModel
25
 
26
- # Use the model path provided by the endpoint, or default to HF hub
27
- model_id = path if path else "google/siglip2-so400m-patch14-384"
 
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")