Update app.py

app.py

@@ -1,20 +1,27 @@
-import gradio as gr
-import joblib
-
-# Load model + vectorizer
-model = joblib.load("model.pkl")
-vectorizer = joblib.load("vectorizer.pkl")
-
-def predict_category(text):
-    X_emb = vectorizer.transform([text])
-    pred = model.predict(X_emb)[0]
-    return pred
-
-demo = gr.Interface(
-    fn=predict_category,
-    inputs=gr.Textbox(lines=2, placeholder="Enter a headline..."),
-    outputs="text",
-    title="📰 News Headline Classifier",
-    description="Type a news headline and get its predicted category."
-)
-
+import gradio as gr
+import joblib
+
+
+# Load model + vectorizer
+model = joblib.load("model.pkl")
+vectorizer = joblib.load("vectorizer.pkl")
+
+#import skops.io as sio
+#model = sio.load("model.skops")
+#vectorizer = sio.load("vectorizer.skops")
+
+def predict_category(text):
+    X_emb = vectorizer.transform([text])
+    pred = model.predict(X_emb)[0]
+    return pred
+
+demo = gr.Interface(
+    fn=predict_category,
+    inputs=gr.Textbox(lines=2, placeholder="Enter a headline..."),
+    outputs="text",
+    title="📰 News Headline Classifier",
+    description="Type a news headline and get its predicted category."
+)
+
+if __name__ == "__main__":
+    demo.launch()