Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
-from transformers import pipeline, AutoConfig, AutoModel
+from transformers import pipeline, AutoConfig, AutoModelForSequenceClassification
 
 model_id = "KenLumod/ML-Fake-Real-News-Detector-Final"
 
 # Force reload model with updated config
 config = AutoConfig.from_pretrained(model_id)
-config.id2label = {1: "Fake News", 0: "Real News"}  # Force override
+config.id2label = {0: "Fake News", 1: "Real News"}  # Force override
 config.label2id = {v: k for k, v in config.id2label.items()}
 
-# Load the model using AutoModel
-model = AutoModel.from_pretrained(model_id, config=config)
+# Load the model using AutoModelForSequenceClassification
+model = AutoModelForSequenceClassification.from_pretrained(model_id, config=config)
 
 # Create the pipeline for classification
 classifier = pipeline(
@@ -21,6 +21,7 @@ classifier = pipeline(
 
 def classify_news(text):
     result = classifier(text)[0]
+    print(result)  # Inspect the output
     return result['label']
 
 demo = gr.Interface(