lfernandopg commited on
Commit
3ee8cff
·
1 Parent(s): f1b1907

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
 
6
  @st.cache(allow_output_mutation=True)
7
  def get_model():
8
- tokenizer = DistilBertTokenizer.from_pretrained('lfernandopg/Proyecto-Transformers')
9
  model = TFDistilBertForSequenceClassification.from_pretrained("lfernandopg/Proyecto-Transformers")
10
  return tokenizer,model
11
 
 
5
 
6
  @st.cache(allow_output_mutation=True)
7
  def get_model():
8
+ tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
9
  model = TFDistilBertForSequenceClassification.from_pretrained("lfernandopg/Proyecto-Transformers")
10
  return tokenizer,model
11