Update main.py

main.py

@@ -57,25 +57,25 @@ async def searches(request: Request):
     return response
 
 @app.route("/imagen", methods=["POST"])
-def generate_image():
-    data = request.get_json()
-    prompt = data.get("prompt")
-    model = data.get("model")
-    sampler = data.get("sampler")
-    seed = int(data.get("seed"))
-    neg = data.get("neg")
+async def generate_image(request: Request):
+    data = await request.json()
+    prompt = data['prompt']
+    model = data.get["model"]
+    sampler = data.get["sampler"]
+    seed = int(data.get["seed"])
+    neg = data.get["neg"]
 
     response = generate_image_prodia(prompt, model, sampler, seed, neg)
     return jsonify({"image": response})
 
 @app.route("/test", methods=["POST"])
-def test():
-    data = request.get_json()
-    prompt = data.get("prompt")
-    model = data.get("model")
-    sampler = data.get("sampler")
-    seed = int(data.get("seed"))
-    neg = data.get("neg")
+def test(request: Request):
+    data = await request.json()
+    prompt = data['prompt']
+    model = data.get["model"]
+    sampler = data.get["sampler"]
+    seed = int(data.get["seed"])
+    neg = data.get["neg"]
 
     res = [prompt, \n, model, \n, sampler, \n, seed, \n, neg]