app.py

import gradio as gr
import torch
from transformers import AutoModelForSequenceClassification, AutoTokenizer
from app.models.predict import predict

app = gr.Interface(fn=predict, inputs="text", outputs="label")
app.launch()