update ui

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from datasets import load_dataset
 from PIL import Image  
 import re
 import os
@@ -14,13 +13,13 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 
 
 model_id = "Next7years/stable-diffusion-v1-5-CatHeiHei-v1"
-device = "cuda"
-#device = "cpu"
+device = torch.device("cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu")
+print("info: running device type: " + device.type )
 
 #word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
 #word_list = word_list_dataset["train"]['text']
 
-
+example_dir = "prompt_examples"
 is_gpu_busy = False
 '''
 def infer(prompt):
@@ -46,12 +45,14 @@ def infer(prompt):
 def infer(prompt):
     samples = 4
     steps = 50
-    scale = 7.5    
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    #pipe = StableDiffusionPipeline.from_pretrained(model_id, device='cpu')
+    scale = 7.5
+    if device.type == "cuda" or device.type == "mps":   
+        pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    else:
+        pipe = StableDiffusionPipeline.from_pretrained(model_id)
     pipe = pipe.to(device)
     images = []
-    results = pipe(prompt, num_images_per_prompt=samples, num_inference_steps=steps, guidance_scale=scale).images
+    results = pipe(prompt, negative_prompt=default_negative_prompt, num_images_per_prompt=samples, num_inference_steps=steps, guidance_scale=scale).images
     print(results)
     for image in results:
         jpeg_image = io.BytesIO()
@@ -187,42 +188,27 @@ css = """
 
 block = gr.Blocks(css=css)
 
-examples = [
-    [
-        'The spirit of a tamagotchi wandering in the city of Paris',
-#        4,
-#        45,
-#        7.5,
-#        1024,
-    ],
-    [
-        'A delicious ceviche cheesecake slice',
-#        4,
-#        45,
-#        7,
-#        1024,
-    ],
-    [
-        'A pao de queijo foodcart in front of a japanese castle',
-#        4,
-#        45,
-#        7,
-#        1024,
-    ],
-    [
-        'alone in the amusement park by Edward Hopper',
-#        4,
-#        45,
-#        7,
-#        1024,
-    ],
-    [
-        "A large cabin on top of a sunny mountain in the style of Dreamworks, artstation",
-#        4,
-#        45,
-#        7,
-#        1024,
-    ],
+def read_files_from_directory(directory):
+    file_contents = []
+    for filename in os.listdir(directory):
+        if filename.endswith(".txt"):
+            file_path = os.path.join(directory, filename)
+            with open(file_path, 'r') as f:
+                content = f.read()
+                file_contents.append([content])
+    return file_contents
+
+examples = read_files_from_directory(example_dir)
+
+metadata = [
+    {"title": "Positive Example",
+     "description": "A positive example input.",
+     "thumbnail": "https://example.com/images/positive.jpg",
+     "label": "Positive"},
+    {"title": "Negative Example",
+     "description": "A negative example input.",
+     "thumbnail": "https://example.com/images/negative.jpg",
+     "label": "Negative"}
 ]
 
 
@@ -238,14 +224,21 @@ with block:
                   font-size: 1.75rem;
                 "
               >
-                <h1 style="font-weight: 900; margin-bottom: 7px;">
-                  CatHeiHei Stable Diffusion v1.5 Model
+                <h1 style="font-weight: 900; margin-bottom: 12px;">
+                   Welcome to CatHeiHei v1 Model
                 </h1>
               </div>
-              <p style="margin-bottom: 10px; font-size: 94%">
-                Stable Diffusion v1-5 is the latest version of the state of the art text-to-image model.<br>For faster generation you can try
-      <a href="https://app.runwayml.com/ai-tools/text-to-image"
-        style="text-decoration: underline;" target="_blank">text to image tool at Runway.</a>
+              <p style="margin-bottom: 10px">
+                 
+                We're excited to open-source this unique AI model, designed specifically to generate images of the world-famous Cat HeiHei. 
+                Our goal is to foster creativity and collaboration within the community, and we can't wait to see the amazing artwork you'll create!
+       
+                <p>Follow us on Instagram: 
+                    <a href="https://www.instagram.com/cat_heihei/" style="display: inline;">
+                        <img src="https://www.instagram.com/static/images/ico/favicon-192.png/68d99ba29cc8.png" alt="Instagram Logo" style="display: inline; width: 20px; height: 20px; margin-right: 5px;">
+                        @cat_heihei
+                    </a>
+                </p>
               </p>
             </div>
         """
@@ -296,7 +289,7 @@ with block:
                 randomize=True,
             )
 
-        ex = gr.Examples(examples=examples, fn=infer, inputs=text, outputs=[gallery], cache_examples=False, postprocess=False)
+        ex = gr.Examples(examples=examples, label="Example prompt to generate CatHeiHei", fn=infer, inputs=text, outputs=[gallery], cache_examples=False, postprocess=False)
         ex.dataset.headers = [""]
 
         text.submit(infer, inputs=text, outputs=[gallery], postprocess=False)
@@ -320,16 +313,15 @@ with block:
         )
         gr.HTML(
             """
-                <div class="footer">
-                    <p>Model by <a href="https://huggingface.co/CompVis" style="text-decoration: underline;" target="_blank">CompVis</a> and <a href="https://runwayml.com/" style="text-decoration: underline;" target="_blank">Runway</a> supported by <a href="https://huggingface.co/stabilityai" style="text-decoration: underline;" target="_blank">Stability AI</a> - backend running JAX on TPUs due to generous support of <a href="https://sites.research.google/trc/about/" style="text-decoration: underline;" target="_blank">Google TRC program</a> - Gradio Demo by 🤗 Hugging Face
+                <div class="footer" style="padding-top: 20px;">
+                    <p>Thanks for using the <b>Cat HeiHei </b> Customized Model! We appreciate your support and creativity. Share your feedback, suggestions, and content ideas by messaging us on Instagram 
+                    <a href="https://www.instagram.com/cat_heihei/" style="display: inline;">
+                        <img src="https://www.instagram.com/static/images/ico/favicon-192.png/68d99ba29cc8.png" alt="Instagram Logo" style="display: inline; width: 20px; height: 20px; margin-right: 5px;">
+                        @cat_heihei
+                    </a>. 
+                    Let's make the Cat HeiHei community a fun, creative space for all! Happy creating! 🐾💖
                     </p>
                 </div>
-                <div class="acknowledgments">
-                    <p><h4>LICENSE</h4>
-The model is licensed with a <a href="https://huggingface.co/spaces/CompVis/stable-diffusion-license" style="text-decoration: underline;" target="_blank">CreativeML Open RAIL-M</a> license. The authors claim no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in this license. The license forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please <a href="https://huggingface.co/spaces/CompVis/stable-diffusion-license" target="_blank" style="text-decoration: underline;" target="_blank">read the license</a></p>
-                    <p><h4>Biases and content acknowledgment</h4>
-Despite how impressive being able to turn text into image is, beware to the fact that this model may output content that reinforces or exacerbates societal biases, as well as realistic faces, pornography and violence. The model was trained on the <a href="https://laion.ai/blog/laion-5b/" style="text-decoration: underline;" target="_blank">LAION-5B dataset</a>, which scraped non-curated image-text-pairs from the internet (the exception being the removal of illegal content) and is meant for research purposes. You can read more in the <a href="https://huggingface.co/runwayml/stable-diffusion-v1-5" style="text-decoration: underline;" target="_blank">model card</a></p>
-               </div>
            """
         )
 

prompt_examples/catheihei_baking.txt

@@ -0,0 +1 @@
+Create a photo-realistic image of CatHeiHei cat making bakery treats while wearing a chef hat. CatHeiHei cat should have white, orange, and black calico fur patterns and yellow eyes. The cat should be sitting on a kitchen counter, surrounded by baking ingredients and utensils, with its front paws playfully handling the dough. The setting should be a warm, inviting kitchen filled with delicious baked goods. The image must have a realistic, photographic style, avoiding any anime or cartoon-like elements

prompt_examples/catheihei_bathing.txt

@@ -0,0 +1,5 @@
+Create a photo-realistic image of CatHeiHei cat taking a delightful bath at night, outdoors under a starry sky with a visible moon. CatHeiHei cat should have white, orange, and black calico fur patterns and captivating yellow eyes. The cat should be sitting or partially submerged in a warm, bubbly bathtub, with steam or hot water aura rising from the surface.
+
+Surround the bathtub with gently flickering candles, casting a warm and soothing glow on HeiHei and the immediate surroundings. The focus should be on CatHeiHei cat's adorable expressions and relaxed demeanor, capturing the essence of a serene and enjoyable bathing experience. The background should feature a clear, star-filled sky with a bright moon illuminating the scene.
+
+The image must have a realistic, photographic style, avoiding any anime or cartoon-like elements.

prompt_examples/catheihei_winebrand.txt

@@ -0,0 +1 @@
+Create a photo-realistic image of CatHeiHei cat lean against a bottle of wine. (face on the label of the bottle), CatHeiHei cat should have white, orange, and black calico fur patterns and yellow eyes. The cat should be sitting upright, gently holding the wine bottle with its front paws. The setting should be elegant, with a dimly lit background, creating a cozy atmosphere. The image must have a realistic, photographic style, avoiding any anime or cartoon-like elements.

requirements.txt

@@ -8,4 +8,6 @@ numpy
 matplotlib
 uuid
 opencv-python
+accelerate
+gradio
 git+https://github.com/openai/CLIP.git