Update handler.py
Browse files- 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)
|