Hugging Face's logo

Spaces:
karths
/
types_issues
Running on Zero

App Files Community
karths commited on
Commit
112c46b
·
verified ·
1 Parent(s): 888f8f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ quality_mapping = {
52
  }
53
 
54
  # Pre-load models and tokenizer for quality prediction
55
- tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
56
  models = {path: AutoModelForSequenceClassification.from_pretrained(path) for path in model_paths}
57
 
58
  def get_quality_name(model_name):
 
52
  }
53
 
54
  # Pre-load models and tokenizer for quality prediction
55
+ tokenizer = AutoTokenizer.from_pretrained("distilbert/distilroberta-base")
56
  models = {path: AutoModelForSequenceClassification.from_pretrained(path) for path in model_paths}
57
 
58
  def get_quality_name(model_name):