Spaces:
Sleeping
Sleeping
import gradio as gr | |
from load_model import load_model, predict_probs | |
from constants import title, description, examples | |
model = load_model() | |
demo = gr.Interface( | |
fn=lambda text: predict_probs(model, text), | |
inputs=gr.Textbox(label='News article title and description'), | |
outputs=gr.Label(num_top_classes=4), | |
examples=examples, | |
allow_flagging='never', | |
title=title, | |
description=description | |
) | |
demo.launch() | |