Update app.py

app.py

@@ -22,8 +22,8 @@ from fastapi import FastAPI
 app = FastAPI()
 @app.get("/")
 
-def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cfg_scale=7, seed=-1, width=896, height=1152):
-    result = {prompt:"prompt", negative_prompt:"negative_prompt", task:"task", steps:"steps", sampler:"sampler", cfg_scale:"cfg_scale", seed:"seed" , width:"width" , height:"height" }
+def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cfg_scale=7, seed=-1, style_name, width=896, height=1152):
+    result = {prompt:"prompt", negative_prompt:"negative_prompt", task:"task", steps:"steps", sampler:"sampler", cfg_scale:"cfg_scale", seed:"seed" , style_name:"style_name", width:"width" , height:"height" }
     print(result)
     try:
         language = detect(prompt)
@@ -41,7 +41,7 @@ def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cf
     seed = int(seed)
     width = int(width)
     height = int(height)
-    style_name= int(style_name)
+    style_name = int(style_name)
     progress=gr.Progress(track_tqdm=True)
 
     if task == "Playground v2":