import gradio as gr
from joblib import load


model = load('model.pkl')


def predict_sentiment(text: str)-> str:

    
    sentiment_map = {
        -1: 'Negative',
        0: 'Neutral',
        1: 'Positive'
    }

    prediction = model.predict([text])[0]
    return sentiment_map[prediction]

demo = gr.Interface(fn=predict_sentiment, inputs="text", outputs="text")
demo.launch()