Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
-from transformers import BertTokenizer, BertForSequenceClassification
+from transformers import BertTokenizer, TFAutoModelForSequenceClassification
 
 @st.cache(allow_output_mutation=True)
 def get_model():
     tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-    model = BertForSequenceClassification.from_pretrained('PlengP/hfphishoffbert300000', from_tf=True)
+    model = TFAutoModelForSequenceClassification.from_pretrained('PlengP/hfphishoffbert300000')
     return tokenizer, model
 
 tokenizer, model = get_model()