JanviMl commited on
Commit
1f2d2e0
·
verified ·
1 Parent(s): 5bdb39c

Update model_loader.py

Browse files
Files changed (1) hide show
  1. model_loader.py +1 -1
model_loader.py CHANGED
@@ -7,7 +7,7 @@ def load_model_and_tokenizer():
7
  Returns the model and tokenizer for use in classification.
8
  """
9
  try:
10
- model_name = "your_username/xlm-roberta-toxic-classifier" # Replace with your model repo ID
11
  # If the model is local: model_name = "./model"
12
 
13
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
7
  Returns the model and tokenizer for use in classification.
8
  """
9
  try:
10
+ model_name = "JanviMl/xlm-roberta-toxic-classifier-capstone" # Replace with your model repo ID
11
  # If the model is local: model_name = "./model"
12
 
13
  model = AutoModelForSequenceClassification.from_pretrained(model_name)