Spaces:
Running
Running
# Imports | |
import gradio as gr | |
from sklearn.linear_model import LogisticRegression | |
import pickle5 as pickle | |
# Load model from pickle file | |
model = pickle.load(open('lg_classifier.sav', 'rb')) | |
# Define function to predict | |
def predict(text): | |
return model.predict([text])[0] | |
# Define interface | |
iface = gr.Interface(fn=predict, | |
title="Text Classification Demo", | |
description="This is a demo of a text classification model using Logistic Regression.", | |
inputs=gr.Textbox(lines=10, placeholder='Input text here...', label="Input Text"), | |
outputs=gr.Textbox(label="Predicted Label", lines=2, placeholder='Predicted label will appear here...'), | |
examples=['The indictments were announced Tuesday by the Justice Department in Cairo.', "In 2019, the men's singles winner was Novak Djokovic who defeated Roger Federer in a tournament taking place in the United Kingdom.", 'In a study published by the American Heart Association on January 18, researchers at the Johns Hopkins School of Medicine found that meal timing did not impact weight.'], | |
allow_flagging='never' | |
) | |
demo = iface.launch(share=True) | |