Update app.py

app.py

@@ -23,7 +23,7 @@ classes = ['pt', 'pt_br']
 class_label = ClassLabel(names=classes)
 
 
-@st.cache_data
+@st.cache
 def get_model():
     return BertForSequenceClassification.from_pretrained(
    './pt_br_model',