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()