Update main.py

main.py

@@ -73,16 +73,10 @@ class User(BaseModel):
     model: str
     sampler: str
     seed: int
-    neg: str or None
+    neg: str | None = None
 
 @app.route("/test", methods=["POST"])
 async def test(request: User):
-    data = {
-        'prompt': request.prompt,
-        'model': request.model,
-        'sampler': request.sampler,
-        'seed': request.seed,
-        'neg': request.neg
-    }
-    return data
+    
+    return request