Update main.py

main.py

@@ -145,8 +145,12 @@ details = {
 def image_detail(detail_id: int = Path(None, description="The ID of 1.model id and 2.sampler id", gt=0, lt=3)):
     return details[detail_id]
 
+class Test(BaseModel):
+    prompt: str
+    model: str = 'gpt-3.5-turbo'
+    neg: str = False
 
 @app.post("/test")
-def test(request: User):
-        return {'data': f'Prompt is {request.prompt} Model is {request.model}'}
+def test(request: Test):
+        return {'data': f'Prompt is {request.prompt} Model is {request.model} Neg is {request.neg}'}