Update app.py

app.py

@@ -11,15 +11,14 @@ from Scraper import Scrap
 st.set_page_config(layout="wide")
 
 model_checkpoint = "Rifky/FND"
-data_checkpoint = os.path.join(os.getcwd(), "data", "turnbackhoax.csv")
+data_checkpoint = "Rifky/turnbackhoax-encoded"
 label = {0: "valid", 1: "fake"}
 
 @st.cache(show_spinner=False, allow_output_mutation=True)
 def load_model():
     model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
-    base_model = AutoModel.from_pretrained(model_checkpoint)
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, fast=True)
-    return model, base_model, tokenizer
+    return model, tokenizer
 
 def sigmoid(x):
     return 1 / (1 + np.exp(-x))
@@ -28,7 +27,7 @@ input_column, reference_column = st.columns(2)
 input_column.write('# Fake News Detection AI')
 
 with st.spinner("Loading Model..."):
-    model, base_model, tokenizer = load_model()
+    model, tokenizer = load_model()
     data = pd.read_csv(data_checkpoint)
 
 user_input = input_column.text_input("Article url")