Update app.py

app.py

@@ -37,7 +37,7 @@ if torch.cuda.is_available():
 
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
 
-def on_model_change(model):
+def inference(model, prompt, guidance, steps):
 
     global current_model
     global pipe
@@ -47,13 +47,18 @@ def on_model_change(model):
         if torch.cuda.is_available():
             pipe = pipe.to("cuda")
 
-def inference(prompt, guidance, steps):
-
     prompt = prompt_prefixes[current_model] + prompt
     image = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512).images[0]
     return image
 
-with gr.Blocks() as demo:
+css = """
+  <style>
+  a {
+    text-decoration: underline;
+  }
+  </style>
+"""
+with gr.Blocks(css=css) as demo:
     gr.HTML(
         """
             <div style="text-align: center; max-width: 700px; margin: 0 auto;">
@@ -70,7 +75,7 @@ with gr.Blocks() as demo:
                 </h1>
               </div>
               <p style="margin-bottom: 10px; font-size: 94%">
-               Demo for multiple fine-tuned Stable Diffusion models, trained on different styles:
+               Demo for multiple fine-tuned Stable Diffusion models, trained on different styles: <br>
                <a href="https://huggingface.co/nitrosocke/Arcane-Diffusion">Arcane</a>, <a href="https://huggingface.co/nitrosocke/archer-diffusion">Archer</a>, <a href="https://huggingface.co/nitrosocke/elden-ring-diffusion">Elden Ring</a>, <a href="https://huggingface.co/nitrosocke/spider-verse-diffusion">Spiderverse</a>, <a href="https://huggingface.co/nitrosocke/modern-disney-diffusion">Modern Disney</a>, <a href="https://huggingface.co/hakurei/waifu-diffusion">Waifu</a>, <a href="https://huggingface.co/lambdalabs/sd-pokemon-diffusers">Pokemon</a>, <a href="https://huggingface.co/yuk/fuyuko-waifu-diffusion">Fuyuko Waifu</a>, <a href="https://huggingface.co/AstraliteHeart/pony-diffusion">Pony</a>, <a href="https://huggingface.co/IfanSnek/JohnDiffusion">John</a>, <a href="https://huggingface.co/nousr/robo-diffusion">Robo</a>.
               </p>
             </div>
@@ -88,18 +93,17 @@ with gr.Blocks() as demo:
         with gr.Column():
             image_out = gr.Image(height=512)
 
-    model.change(on_model_change, inputs=model, outputs=[])
-    run.click(inference, inputs=[prompt, guidance, steps], outputs=image_out)
+    run.click(inference, inputs=[model, prompt, guidance, steps], outputs=image_out)
     gr.Examples([
-        ["jason bateman disassembling the demon core", 7.5, 50],
-        ["portrait of dwayne johnson", 7.0, 75],
-        ["portrait of a beautiful alyx vance half life", 10, 50],
-        ["Aloy from Horizon: Zero Dawn, half body portrait, smooth, detailed armor, beautiful face, illustration", 7, 45],
-        ["fantasy portrait painting, digital art", 4, 30],
+        [models[0], "jason bateman disassembling the demon core", 7.5, 50],
+        [models[3], "portrait of dwayne johnson", 7.0, 75],
+        [models[4], "portrait of a beautiful alyx vance half life", 10, 50],
+        [models[5], "Aloy from Horizon: Zero Dawn, half body portrait, smooth, detailed armor, beautiful face, illustration", 7, 45],
+        [models[4], "fantasy portrait painting, digital art", 4, 30],
     ], [prompt, guidance, steps], image_out, inference, cache_examples=torch.cuda.is_available())
     gr.HTML('''
         <div>
-            <p>Model by <a href="https://huggingface.co/nitrosocke" style="text-decoration: underline;" target="_blank">@nitrosocke</a> ❤️</p>
+            <p>Model by <a href="https://huggingface.co/nitrosocke" target="_blank">@nitrosocke</a> ❤️</p>
         </div>
         <div>Space by 
             <a href="https://twitter.com/hahahahohohe">