kalebu's picture
solved identation issue
4b65d0c
raw
history blame contribute delete
616 Bytes
import joblib
import gradio as gr
model = joblib.load("model.pkl")
def get_sentiment(input_text):
predictions = model.predict_proba([input_text])
label = {
"negative":str(predictions[0][0]),
"positive":str(predictions[0][1]),
}
return label
iface = gr.Interface(
fn=get_sentiment,
inputs=[gr.Text(text="Sijapendezwa na tabia yake")],
outputs=[gr.Label(label='Sentiment')],
title="Swahili Sentiment Analysis using",
description="Swahili sentiment analysis is based on a Bayesian scikit-learn model"
)
iface.launch(inline=False)