Update main.py

main.py

@@ -77,11 +77,7 @@ class User(BaseModel):
 
 @app.post("/test")
 async def test(request: User):
-        prompt = request.prompt
-        model = request.model
-        sampler = request.sampler
-        seed = request.seed
-        neg = request.neg
-        res = f'Prompt: {prompt}'
-    return res
+        data = await request.json()
+        prompt = data['prompt']
+        return data