Zen0 commited on
Commit
b076813
·
verified ·
1 Parent(s): e26d773

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -62,7 +62,7 @@ async def evaluate_text(request: TextEvaluationRequest):
62
 
63
 
64
  # Model and Tokenizer
65
- model_name = "mobilebert" # The base model you fine-tuned from
66
  tokenizer = AutoTokenizer.from_pretrained(model_name)
67
  model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
68
 
 
62
 
63
 
64
  # Model and Tokenizer
65
+ model_name = "Zen0/FrugalDisinfoHunter"
66
  tokenizer = AutoTokenizer.from_pretrained(model_name)
67
  model = AutoModelForSequenceClassification.from_pretrained("Zen0/FrugalDisinfoHunter", num_labels=8)
68