Update app.py

app.py

@@ -43,12 +43,25 @@ st.caption("Multi-Modal Fake News & AI Image Detection System")
 # LOAD MODELS
 # ---------------------------
 
-# TEXT MODEL (DistilBERT)
+# # TEXT MODEL (DistilBERT)
+# @st.cache_resource
+# def load_text_model():
+#     model_name = "Maheentouqeer1/truthguard-fake-news-detector"
+#     tokenizer = AutoTokenizer.from_pretrained(model_name)
+#     model = AutoModelForSequenceClassification.from_pretrained(model_name)
+#     return tokenizer, model
 @st.cache_resource
 def load_text_model():
+    from transformers import AutoTokenizer, AutoModelForSequenceClassification
+    
     model_name = "Maheentouqeer1/truthguard-fake-news-detector"
+    
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSequenceClassification.from_pretrained(model_name)
+    model = AutoModelForSequenceClassification.from_pretrained(
+        model_name,
+        low_cpu_mem_usage=True
+    )
+    
     return tokenizer, model
 
 # # IMAGE MODEL
@@ -97,6 +110,11 @@ def predict_news(text):
     return prediction, confidence
 
 # ---------------------------
+with st.spinner("Loading models... ⏳ Please wait"):
+    tokenizer, text_model = load_text_model()
+    image_model = load_image_model()
+
+    st.success("Models Loaded Successfully ✅")
 # TABS
 # ---------------------------
 tab1, tab2 = st.tabs(["📰 Fake News Detection", "🖼️ AI Image Detection"])