Update main.py

main.py

@@ -72,10 +72,10 @@ async def generate_image(request: Request):
 async 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"]
+    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