test

main.py

@@ -1,7 +1,7 @@
 # https://medium.com/@qacheampong/building-and-deploying-a-fastapi-app-with-hugging-face-9210e9b4a713
 # https://huggingface.co/spaces/Queensly/FastAPI_in_Docker
 
-from fastapi import FastAPI
+from fastapi import FastAPI,Request
 import uvicorn
 
 app = FastAPI()
@@ -13,9 +13,8 @@ app = FastAPI()
 def root():
     return {"API": "Sum of 2 Squares"}
     
-@app.route('/predict', methods=['POST']) 
-def predict():
-
+@app.post("/predict")
+def predict(request: Request):
   prompt = request.json.get('prompt')
   return "你好啊"+prompt