Update main.py

main.py

@@ -71,12 +71,5 @@ async def generate_image(request: Request):
 @app.route("/test", methods=["POST"])
 async def test(request: Request):
     data = await request.json()
-    prompt = data['prompt']
-    model = data["model"]
-    sampler = data["sampler"]
-    seed = int(data["seed"])
-    neg = data["neg"]
-
-    res = f'prompt: {prompt} \n model: {model} \n sampler: {sampler} \n seed: {seed} \n neg: {neg}'
-    return res
+    return data