rroset commited on
Commit
c0a7220
verified
1 Parent(s): 920bc12

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -8,8 +8,8 @@ from typing import Dict, List, Any
8
  class EndpointHandler():
9
  def __init__(self, path=""):
10
  # Utilitza el nom correcte del model per carregar-lo des de Hugging Face Hub
11
- self.model = CLIPModel.from_pretrained("rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
12
- self.processor = CLIPProcessor.from_pretrained("rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
13
 
14
  # Crea la pipeline de classificaci贸 d'imatges zero-shot
15
  self.classifier = pipeline("zero-shot-image-classification", model=self.model, processor=self.processor)
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
10
  # Utilitza el nom correcte del model per carregar-lo des de Hugging Face Hub
11
+ self.model = CLIPModel.from_pretrained("hf-hub:rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
12
+ self.processor = CLIPProcessor.from_pretrained("hf-hub:rroset/CLIP-ViT-B-32-laion2B-s34B-b79K")
13
 
14
  # Crea la pipeline de classificaci贸 d'imatges zero-shot
15
  self.classifier = pipeline("zero-shot-image-classification", model=self.model, processor=self.processor)