Update app.py

app.py

@@ -1,10 +1,24 @@
 import gradio as gr
 import joblib
+from huggingface_hub import hf_hub_download
 
-# Load model + vectorizer
-model = joblib.load("clickbait_detector.pkl")
-vectorizer = joblib.load("tfidf_vectorizer.pkl")
+# Download model + vectorizer from your HF repo (public repo)
+model_path = hf_hub_download(
+    repo_id="Devishetty100/clickbait-detector",
+    filename="clickbait_detector.pkl",
+    token=None  # None for public repo
+)
+vectorizer_path = hf_hub_download(
+    repo_id="Devishetty100/clickbait-detector",
+    filename="tfidf_vectorizer.pkl",
+    token=None
+)
+
+# Load them
+model = joblib.load(model_path)
+vectorizer = joblib.load(vectorizer_path)
 
+# Prediction function
 def predict_headline(headline):
     pred = model.predict(vectorizer.transform([headline]))[0]
     return pred