Spaces:

lfernandopg
/

Proyecto-Transformers

Sleeping

lfernandopg commited on Mar 31, 2023

Commit

4edd498

1 Parent(s): 4554bb8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import tensorflow as tf
 @st.cache(allow_output_mutation=True)
 def get_model():
-    tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
     model = TFDistilBertForSequenceClassification.from_pretrained("lfernandopg/Proyecto-Transformers")
     return tokenizer,model

 @st.cache(allow_output_mutation=True)
 def get_model():
+    tokenizer = DistilBertTokenizer.from_pretrained('lfernandopg/Proyecto-Transformers')
     model = TFDistilBertForSequenceClassification.from_pretrained("lfernandopg/Proyecto-Transformers")
     return tokenizer,model