test

main.py

@@ -3,6 +3,7 @@
 
 from fastapi import FastAPI,Request
 import uvicorn
+import json
 
 app = FastAPI()
 
@@ -13,10 +14,17 @@ app = FastAPI()
 def root():
     return {"API": "Sum of 2 Squares"}
     
+@app.post("/img2img")
+async def predict(url:str,prompt:str):
+  body = await request.body()
+  data = json.loads(body)
+  prompt = data.get("prompt")
+  return f"您好,{url+prompt}"    
+    
 @app.post("/predict")
-async def predict(request: Request):
-  data = await request.json()
+async def predict(request:Request):
+  body = await request.body()
+  data = json.loads(body)
   prompt = data.get("prompt")
   return f"您好,{prompt}"
 
-