test

main.py

@@ -13,10 +13,10 @@ app = FastAPI()
 def root():
     return {"API": "Sum of 2 Squares"}
     
-@app.post('/predict')
-async def predict_with_prompt(prompt: str):
+@app.route('/predict', methods=['POST']) 
+def predict():
 
-  
-  return "你好"+prompt
+  prompt = request.json.get('prompt')
+  return "你好啊"+prompt