build commit

README.md

@@ -4,10 +4,12 @@ emoji: 🐠
 colorFrom: yellow
 colorTo: purple
 sdk: gradio
-sdk_version: 4.26.0
+sdk_version: 4.25.0
 app_file: app.py
-pinned: false
+pinned: true
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# awesome-tiny-sd
+
+Tiny stable diffusion chatbot based on [segmind/small-sd](https://huggingface.co/segmind/small-sd) to generate image upon text prompts.

app.py

@@ -0,0 +1,62 @@
+import gradio as gr
+import time
+from imgen import *
+
+def print_like_dislike(x: gr.LikeData):
+    print(x.index, x.value, x.liked)
+
+def add_message(history, message):
+    if len(message["files"]) > 0:
+        history.append((message["files"], None))
+    if message["text"] is not None and message["text"] != "":
+        history.append((message["text"], None))
+    return history, gr.MultimodalTextbox(value=None, interactive=False)
+
+
+
+def bot(history):
+    if type(history[-1][0]) != tuple:
+        try:
+            prompt = history[-1][0]
+            image = pipeline(prompt).images[0]
+            image.save("generated_image.png")
+            response = ("generated_img.png",)
+            history[-1][1] = response
+            yield history
+        except Exception as e:
+            response = f"Sorry, the error '{e}' occured while generating the response; check [troubleshooting documentation](https://astrabert.github.io/awesome-tiny-sd/#troubleshooting) for more"
+            history[-1][1] = ""
+            for character in response:
+                history[-1][1] += character
+                time.sleep(0.05)
+                yield history
+    if type(history[-1][0]) == tuple:
+        response = f"Sorry, this version still does not support uploaded files :("
+        history[-1][1] = ""
+        for character in response:
+            history[-1][1] += character
+            time.sleep(0.05)
+            yield history
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(
+        [[None, ("Hi, I am awesome-tiny-sd, a little stable diffusion model that lets you generate images:blush:\nJust write me a prompt, I'll generate what you ask for:heart:",)]],
+        label="awesome-tiny-sd",
+        elem_id="chatbot",
+        bubble_full_width=False,
+    )
+
+    chat_input = gr.MultimodalTextbox(interactive=True, file_types=["pdf"], placeholder="Enter your image-generating prompt...", show_label=False)
+
+    chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
+    bot_msg = chat_msg.then(bot, chatbot, chatbot, api_name="bot_response")
+    bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
+
+    chatbot.like(print_like_dislike, None, None)
+    clear = gr.ClearButton(chatbot)
+
+demo.queue()
+if __name__ == "__main__":
+    demo.launch()
+
+	

imgen.py

@@ -0,0 +1,4 @@
+from diffusers import DiffusionPipeline
+import torch
+
+pipeline = DiffusionPipeline.from_pretrained("segmind/small-sd", torch_dtype=torch.float32)

requirements.txt

@@ -0,0 +1,3 @@
+gradio==4.25.0
+diffusers==0.27.2
+torch==2.1.2