Zen0 commited on
Commit
75b7bad
·
verified ·
1 Parent(s): f8fc304

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -64,7 +64,7 @@ async def evaluate_text(request: TextEvaluationRequest):
64
  # Model and Tokenizer
65
  model_name = "google/mobilebert-uncased" # The base model you fine-tuned from
66
  tokenizer = AutoTokenizer.from_pretrained(model_name)
67
- model = AutoModelForSequenceClassification.from_pretrained("./pytorch_model.bin", num_labels=8)
68
 
69
  # Tokenize the test data
70
  test_texts = test_dataset["quote"]
 
64
  # Model and Tokenizer
65
  model_name = "google/mobilebert-uncased" # 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
 
69
  # Tokenize the test data
70
  test_texts = test_dataset["quote"]