Update app.py

app.py

@@ -1,12 +1,12 @@
 import streamlit as st
 import numpy as np
-from transformers import BertTokenizer, TFBertForSequenceClassification
+from transformers import DistilBertTokenizer, TFDistilBertForSequenceClassification
 import torch
 
 @st.cache(allow_output_mutation=True)
 def get_model():
-    tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-    model = TFBertForSequenceClassification.from_pretrained("lfernandopg/Proyecto-Transformers")
+    tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
+    model = TFDistilBertForSequenceClassification.from_pretrained("lfernandopg/Proyecto-Transformers")
     return tokenizer,model