代码初始化

app.py

@@ -1,6 +1,8 @@
 from fastapi import FastAPI
+from inference.inference import router
 
 app = FastAPI()
+app.include_router(router)
 
 
 @app.get("/")

{inference-api → inference}/apiModel.py

@@ -3,6 +3,5 @@ from __future__ import annotations
 from pydantic import BaseModel
 
 
-class ApiModel(BaseModel):
-    token: str
-    payload: dict | None = None
+class Payload(BaseModel):
+    input: str | None = None

{inference-api → inference}/inference.py

@@ -1,22 +1,22 @@
 import requests
-from fastapi import FastAPI, Header
-from apiModel import ApiModel
+from fastapi import APIRouter, Header
+from apiModel import Payload
 
-app = FastAPI()
+router = APIRouter()
 
 API_URL = "https://api-inference.huggingface.co/models/cardiffnlp/twitter-roberta-base-sentiment-latest"
 
 
-@app.post("/api-inference/")
+@router.post("/api-inference/")
 async def api_inference(
         authorization: str = Header(...),
-        item: ApiModel = None):
-    print("请求：",item)
+        item: Payload = None):
+    print("请求：", item)
     headers = {"Authorization": authorization}
     # Forward request to Hugging Face API
     response = requests.post(
         API_URL,
         headers=headers,
-        json=item.payload.dict()
+        json=item
     )
     return response.json()