代码初始化

inference-api/inference.py

@@ -1,5 +1,5 @@
 import requests
-from fastapi import FastAPI
+from fastapi import FastAPI, Header
 from apiModel import ApiModel
 
 app = FastAPI()
@@ -7,12 +7,16 @@ app = FastAPI()
 API_URL = "https://api-inference.huggingface.co/models/cardiffnlp/twitter-roberta-base-sentiment-latest"
 
 
-@app.post("/api-inference")
-def get_version(item: ApiModel):
-    # payload = {
-    #     "inputs": "Today is a great day",
-    # }
-    print("请求参数:", item)
-    headers = {"Authorization": item.token}
-    response = requests.post(API_URL, headers=headers, json=item.payload)
+@app.post("/api-inference/")
+async def api_inference(
+        authorization: str = Header(...),
+        item: ApiModel = None):
+    print("请求：",item)
+    headers = {"Authorization": authorization}
+    # Forward request to Hugging Face API
+    response = requests.post(
+        API_URL,
+        headers=headers,
+        json=item.payload.dict()
+    )
     return response.json()