Update app.py

app.py

@@ -9,7 +9,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
 dtype = torch.float16 if device == "cuda" else torch.float32
 
-pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/sd-pokemon-diffusers", torch_dtype=dtype)
+pipe = StableDiffusionPipeline.from_pretrained("fffiloni/mr-men-and-little-misses", torch_dtype=dtype)
 pipe = pipe.to(device)
 
 
@@ -136,15 +136,13 @@ with block:
         """
             <div style="text-align: center; max-width: 650px; margin: 0 auto;">
               <div>
-                <img class="logo" src="https://lambdalabs.com/hubfs/logos/lambda-logo.svg" alt="Lambda Logo"
-                    style="margin: auto; max-width: 7rem;">
+                
                 <h1 style="font-weight: 900; font-size: 3rem;">
-                  Pokémon text to image
+                  Mr Men and Little Misses text to image
                 </h1>
               </div>
               <p style="margin-bottom: 10px; font-size: 94%">
-              Generate new Pokémon from a text description,
-                <a href="https://lambdalabs.com/blog/how-to-fine-tune-stable-diffusion-how-we-made-the-text-to-pokemon-model-at-lambda/">created by Lambda Labs</a>.
+              Generate your own Mr men or Little Misses.
               </p>
             </div>
         """
@@ -189,14 +187,11 @@ with block:
         gr.HTML(
             """
                 <div class="footer">
-                    <p> Gradio Demo by 🤗 Hugging Face and Lambda Labs
+                    <p> Gradio Demo by 🤗 @fffiloni
                     </p>
                 </div>
                 <div class="acknowledgments">
-                    <p> Put in a text prompt and generate your own Pokémon character, no "prompt engineering" required!
-                    <p>If you want to find out how we made this model read about it in <a href="https://lambdalabs.com/blog/how-to-fine-tune-stable-diffusion-how-we-made-the-text-to-pokemon-model-at-lambda/">this blog post</a>.
-                    <p>And if you want to train your own Stable Diffusion variants, see our <a href="https://github.com/LambdaLabsML/examples/tree/main/stable-diffusion-finetuning">Examples Repo</a>!
-                    <p>Trained by <a href="justinpinkney.com">Justin Pinkney</a> (<a href="https://twitter.com/Buntworthy">@Buntworthy</a>) at <a href="https://lambdalabs.com/">Lambda Labs</a>.</p>
+                    <p> Put in a text prompt and generate your own Mr Men or Little Misses character, no "prompt engineering" required!</p>
                </div>
            """
         )