x2bee
/

ModernBERT-ecs-GIST-category

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

CocoRoF commited on Mar 6, 2025

Commit

b6bc975

·

verified ·

1 Parent(s): c237072

Update model.py

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import AutoTokenizer, TextClassificationPipeline, AutoModelForSequenceClassification;
 import torch
-class PlateerCategoryClassifier(TextClassificationPipeline):
     def __init__(self, repo_id, power_device=None):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"

 from transformers import AutoTokenizer, TextClassificationPipeline, AutoModelForSequenceClassification;
 import torch
+class MajorClassifier(TextClassificationPipeline):
     def __init__(self, repo_id, power_device=None):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"