Update app.py

app.py

@@ -1,21 +1,11 @@
-# app.py
+import gradio as gr
 
-from fastapi import FastAPI
-from pydantic import BaseModel
-from transformers import pipeline
 
-# 创建 FastAPI 应用
-app = FastAPI()
+def greet(name):
+    return "Hello " + name + "!"
 
-# 加载 Hugging Face 模型，假设是一个文本分类模型
-model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
 
-# 定义请求体格式
-class TextRequest(BaseModel):
-    text: str
+demo = gr.Interface(fn=greet, inputs="textbox", outputs="textbox")
 
-# 定义一个 POST 路由，接收文本并返回分类结果
-@app.post("/predict/")
-async def predict(request: TextRequest):
-    result = model(request.text)
-    return {"label": result[0]['label'], "score": result[0]['score']}
+if __name__ == "__main__":
+    demo.launch()