Update app.py

app.py

@@ -1,21 +1,26 @@
 import gradio as gr
 import re
 
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
-model = "Narrativaai/fake-news-detection-spanish"
-classifier = pipeline("text-classification", model=model)
+ckpt = "Narrativaai/fake-news-detection-spanish"
 
+tokenizer = AutoTokenizer.from_pretrained(ckpt)
 
-def predict(header, text):
+model = AutoModelForSequenceClassification.from_pretrained(ckpt)
+
+classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
+
+
+def prediction(header, text):
         results = classifier(header + " [SEP] " + text)
         return results[0]["label"], round(results[0]["score"], 5)
 
 
 gradio_ui = gr.Interface(
-    fn=predict,
+    fn=prediction,
     title="Fake News Detector (Spanish)",
-    description="Enter some text and check if model detects bullying.",
+    description="Enter some text and check if it is Real or Fake",
     inputs=[
         gr.inputs.Textbox(lines=1, label="Type/Paste your headline here"),
         gr.inputs.Textbox(lines=6, label="Type/Paste the article body here"),