import gradio as gr
from transformers import pipeline
# 加载模型
sentiment_model = pipeline("text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")

def predict(text):
    return sentiment_model(text)

# 构建 UI
interface = gr.Interface(
    fn=predict,
    inputs="text",
    outputs="text",
    title="chatgpt-detector-roberta"
)

interface.launch()