代码初始化

inference-api/inference.py

@@ -7,12 +7,12 @@ app = FastAPI()
 API_URL = "https://api-inference.huggingface.co/models/cardiffnlp/twitter-roberta-base-sentiment-latest"
 
 
-@app.post("/api-inference/")
+@app.post("/api-inference")
 def get_version(item: ApiModel):
     # payload = {
     #     "inputs": "Today is a great day",
     # }
-    print("请求参数:", ApiModel)
-    headers = {"Authorization": ApiModel.token}
-    response = requests.post(API_URL, headers=headers, json=ApiModel.payload)
+    print("请求参数:", item)
+    headers = {"Authorization": item.token}
+    response = requests.post(API_URL, headers=headers, json=item.payload)
     return response.json()