fixes

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "editor.wordWrap": "on"
+}

app.py

@@ -19,7 +19,7 @@ models = [
 ]
 
 last_mode = "txt2img"
-current_model = models[1]
+current_model = models[0]
 current_model_path = current_model.path
 
 if is_colab:
@@ -27,14 +27,14 @@ if is_colab:
 
 else: # download all models
   vae = AutoencoderKL.from_pretrained(current_model.path, subfolder="vae", torch_dtype=torch.float16)
-  for model in models[1:]:
+  for model in models[0:]:
     try:
         unet = UNet2DConditionModel.from_pretrained(model.path, subfolder="unet", torch_dtype=torch.float16)
         model.pipe_t2i = StableDiffusionPipeline.from_pretrained(model.path, unet=unet, vae=vae, torch_dtype=torch.float16)
         model.pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(model.path, unet=unet, vae=vae, torch_dtype=torch.float16)
     except:
         models.remove(model)
-  pipe = models[1].pipe_t2i
+  pipe = models[0].pipe_t2i
   
 if torch.cuda.is_available():
   pipe = pipe.to("cuda")
@@ -171,13 +171,13 @@ with gr.Blocks(css=css) as demo:
         f"""
             <div class="finetuned-diffusion-div">
               <div>
-                <h1>Finetuned Diffusion</h1>
+                <h1>Playground Diffusion</h1>
               </div>
               <p>
                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">Spider-Verse</a>, <a href="https://huggingface.co/nitrosocke/modern-disney-diffusion">Modern Disney</a>, <a href="https://huggingface.co/nitrosocke/classic-anim-diffusion">Classic Disney</a>, <a href="https://huggingface.co/hakurei/waifu-diffusion">Waifu</a>, <a href="https://huggingface.co/lambdalabs/sd-pokemon-diffusers">Pokémon</a>, <a href="https://huggingface.co/AstraliteHeart/pony-diffusion">Pony Diffusion</a>, <a href="https://huggingface.co/nousr/robo-diffusion">Robo Diffusion</a>, <a href="https://huggingface.co/DGSpitzer/Cyberpunk-Anime-Diffusion">Cyberpunk Anime</a>, <a href="https://huggingface.co/dallinmackay/Tron-Legacy-diffusion">Tron Legacy</a> + any other custom Diffusers 🧨 SD model hosted on HuggingFace 🤗.
+               <a href="riccardogiorato/beeple-diffusion">Beeple</a>,
+               Diffusers 🧨 SD model hosted on HuggingFace 🤗.
               </p>
-              <p>Don't want to wait in queue? <a href="https://colab.research.google.com/gist/qunash/42112fb104509c24fd3aa6d1c11dd6e0/copy-of-fine-tuned-diffusion-gradio.ipynb"><img data-canonical-src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" src="https://camo.githubusercontent.com/84f0493939e0c4de4e6dbe113251b4bfb5353e57134ffd9fcab6b8714514d4d1/68747470733a2f2f636f6c61622e72657365617263682e676f6f676c652e636f6d2f6173736574732f636f6c61622d62616467652e737667"></a></p>
                Running on <b>{device}</b>{(" in a <b>Google Colab</b>." if is_colab else "")}
               </p>
             </div>
@@ -233,7 +233,7 @@ with gr.Blocks(css=css) as demo:
     generate.click(inference, inputs=inputs, outputs=image_out)
 
     ex = gr.Examples([
-        [models[1].name, "Neon techno-magic robot with spear pierces an ancient beast, hyperrealism, no blur, 4k resolution, ultra detailed", 7.5, 50],
+        [models[0].name, "Neon techno-magic robot with spear pierces an ancient beast, hyperrealism, no blur, 4k resolution, ultra detailed", 7.5, 50],
     ], [model_name, prompt, guidance, steps, seed], image_out, inference, cache_examples=False)
 
     gr.Markdown('''