AndresR2909
/

finetuning_bert_text_classification

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

Metrics Training metrics Community

AndresR2909 commited on Oct 15, 2024

Commit

a7f3b7d

·

verified ·

1 Parent(s): 4d8a834

Update handler.py

Files changed (1) hide show

handler.py +6 -1

handler.py CHANGED Viewed

@@ -9,7 +9,12 @@ logger = logging.getLogger(__name__)
 class EndpointHandler():
     def __init__(self, path="AndresR2909/finetuning-bert-text-classification"):
         logger.info(f"Modelo cargado desde el path: {path}")
-        self.pipeline = pipeline(model=path,  truncation=True,)
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 class EndpointHandler():
     def __init__(self, path="AndresR2909/finetuning-bert-text-classification"):
         logger.info(f"Modelo cargado desde el path: {path}")
+        # Cargar el pipeline con el modelo especificado
+        try:
+            self.pipeline = pipeline(model=path, truncation=True)
+        except Exception as e:
+            logger.exception(f"Error cargando el modelo desde el path {path}: {e}")
+            raise
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: