Spaces:
Sleeping
Sleeping
File size: 655 Bytes
1d97459 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | 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() |