Update app.py

app.py

@@ -1,17 +1,19 @@
-import gradio as gr
+from fastapi import FastAPI
+from pydantic import BaseModel
 from transformers import pipeline
-# 加载模型
-sentiment_model = pipeline("text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
 
-def predict(text):
-    return sentiment_model(text)
+# 创建 FastAPI 实例
+app = FastAPI()
+
+# 加载预训练模型
+sentiment_model = pipeline("text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
 
-# 构建 UI
-interface = gr.Interface(
-    fn=predict,
-    inputs="text",
-    outputs="text",
-    title="chatgpt-detector-roberta"
-)
+# 定义请求体的格式
+class TextRequest(BaseModel):
+    text: str
 
-interface.launch()
+# 定义一个 POST 请求处理函数
+@app.post("/predict")
+async def predict(request: TextRequest):
+    result = sentiment_model(request.text)
+    return {"result": result}