dappradar
/

setfit-marketplaces-multilabel

Text Classification

sentence-transformers

Inference Endpoints

Model card Files Files and versions Community

Mantas commited on Jun 15, 2023

Commit

ab5a823

•

1 Parent(s): bec438c

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -6,7 +6,7 @@ class EndpointHandler:
         # load model
         self.model = SetFitModel.from_pretrained(path)
-        self.id2label = {{0: 'Art', 1: 'Dedicated', 2: 'Domains', 3: 'Gaming', 4: 'General', 5: 'Music', 6: 'Other'}}
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """

         # load model
         self.model = SetFitModel.from_pretrained(path)
+        self.id2label = {0: 'Art', 1: 'Dedicated', 2: 'Domains', 3: 'Gaming', 4: 'General', 5: 'Music', 6: 'Other'}
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """