Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import pipeline, AutoConfig, AutoModel
-import torch
 
 model_id = "KenLumod/ML-Fake-Real-News-Detector-Final"
 
@@ -9,13 +8,8 @@ config = AutoConfig.from_pretrained(model_id)
 config.id2label = {1: "Fake News", 0: "Real News"}  # Force override
 config.label2id = {v: k for k, v in config.id2label.items()}
 
-# Load the model using AutoModel (ensure safetensors format is handled)
-model = AutoModel.from_pretrained(
-    model_id,
-    config=config,
-    # Specify safe_tensors=True if using safetensors
-    safetensors=True
-)
+# Load the model using AutoModel
+model = AutoModel.from_pretrained(model_id, config=config)
 
 # Create the pipeline for classification
 classifier = pipeline(