dappradar
/

setfit-collections-multilabel

Text Classification

sentence-transformers

text-embeddings-inference

Model card Files Files and versions

Mantas commited on Jun 14, 2023

Commit

ef6e2ec

·

1 Parent(s): 0327902

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: 'Collectibles', 2: 'Domains', 3: 'Games', 4: 'Land', 5: 'Metaverse', 6: 'Music', 7: 'Other', 8: 'PFPs', 9: 'Sports'}}
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """

         # load model
         self.model = SetFitModel.from_pretrained(path)
+        self.id2label = {0: 'Art', 1: 'Collectibles', 2: 'Domains', 3: 'Games', 4: 'Land', 5: 'Metaverse', 6: 'Music', 7: 'Other', 8: 'PFPs', 9: 'Sports'}
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """