Update app.py

app.py

@@ -93,7 +93,8 @@ styles = {
             "text, error, blurry, jpeg artifacts, cropped, worst quality, low quality, normal quality, "
             "jpeg artifacts, signature, watermark, username, artist name, (worst quality, low quality:1.4), "
             "bad anatomy, watermark, signature, text, logo"
-        )
+        ),
+        "steps": 40
     },
     "Vintage Realistic": {
         "prompt": (
@@ -115,7 +116,8 @@ styles = {
             "glitch, deformed, mutated, ugly, disfigured, photoshopped skin, airbrushed skin, glossy skin, canvas frame, "
             "(high contrast:1.2), (over saturated:1.2), (glossy:1.1), cartoon, 3d, disfigured, Photoshop, video game, "
             "ugly, tiling, poorly drawn hands, 3d render, impressionism, digital art"
-        )
+        ),
+        "steps": 30
     },
     "Anime 90's Aesthetic": {
         "prompt": (
@@ -133,7 +135,8 @@ styles = {
             "text, abstract, glitch, deformed, mutated, ugly, disfigured, photoshopped skin, airbrushed skin, glossy skin, "
             "canvas frame, (high contrast:1.2), (over saturated:1.2), (glossy:1.1), disfigured, Photoshop, video game, "
             "ugly, tiling, poorly drawn hands, 3d render, impressionism, eyes, mouth, black skin, pale skin, hair, beard"
-        )
+        ),
+        "steps": 34
     },
     "Anime Style": {
         "prompt": (
@@ -149,7 +152,8 @@ styles = {
             "ugly, disfigured, photoshopped skin, airbrushed skin, glossy skin, canvas frame, (high contrast:1.2), "
             "(over saturated:1.2), (glossy:1.1), cartoon, 3d, disfigured, Photoshop, video game, ugly, tiling, "
             "poorly drawn hands, 3d render, impressionism, digital art"
-        )
+        ),
+        "steps": 28
     },
     "Real 70s": {
         "prompt": (
@@ -170,7 +174,8 @@ styles = {
             "mutated, ugly, disfigured, photoshopped skin, airbrushed skin, glossy skin, canvas frame, (high contrast:1.2), "
             "(over saturated:1.2), (glossy:1.1), cartoon, 3d, disfigured, Photoshop, video game, ugly, tiling, "
             "poorly drawn hands, 3d render, impressionism, digital art"
-        )
+        ),
+        "steps": 40
     }
 }
 
@@ -460,9 +465,9 @@ with gr.Blocks() as demo:
         if evt.index < 0 or evt.index >= len(style_names):
             raise ValueError(f"Invalid index: {evt.index}")
         selected_style = style_names[evt.index]
-        return styles[selected_style]["prompt"], styles[selected_style]["negative_prompt"], selected_style
+        return styles[selected_style]["prompt"], styles[selected_style]["negative_prompt"], styles[selected_style]["steps"], selected_style
 
-    style_images_gallery.select(fn=select_style, inputs=[], outputs=[prompt, negative_prompt, selected_style])
+    style_images_gallery.select(fn=select_style, inputs=[], outputs=[prompt, negative_prompt, num_inference_steps, selected_style])
 
     def update_controlnet(controlnet_type):
         status = load_controlnet_model(controlnet_type)
@@ -547,4 +552,4 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     # Your Gradio interface setup here
     demo.launch(auth=("roland", "roland"), debug=True)
-    clear_memory()
+    clear_memory()