qc7 commited on
Commit
dcbf77d
1 Parent(s): 6c8ab76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import TextClassificationPipeline, AutoTokenizer, AutoModelFor
7
 
8
  @st.cache(suppress_st_warning=True, hash_funcs={transformers.AutoTokenizer: lambda _: None})
9
  def load_tok_and_model():
10
- tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForSequenceClassification.from_pretrained(".")
12
  return tokenizer, model
13
 
 
7
 
8
  @st.cache(suppress_st_warning=True, hash_funcs={transformers.AutoTokenizer: lambda _: None})
9
  def load_tok_and_model():
10
+ tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
11
  model = AutoModelForSequenceClassification.from_pretrained(".")
12
  return tokenizer, model
13