Upload 3 files

app.py

@@ -0,0 +1,56 @@
+# Libraries for building GUI 
+import tkinter as tk
+import customtkinter as ctk 
+
+# Machine Learning libraries 
+import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
+
+# Libraries for processing image 
+from PIL import ImageTk
+
+# private modules 
+from authtoken import auth_token
+
+
+# Create app user interface
+app = tk.Tk()
+app.geometry("532x632")
+app.title("Text to Image app")
+app.configure(bg='black')
+ctk.set_appearance_mode("dark") 
+
+# Create input box on the user interface 
+prompt = ctk.CTkEntry(height=40, width=512, text_font=("Arial", 15), text_color="white", fg_color="black") 
+prompt.place(x=10, y=10)
+
+# Create a placeholder to show the generated image
+img_placeholder = ctk.CTkLabel(height=512, width=512, text="")
+img_placeholder.place(x=10, y=110)
+
+# Download stable diffusion model from hugging face 
+modelid = "CompVis/stable-diffusion-v1-4"
+device = "cuda"
+stable_diffusion_model = StableDiffusionPipeline.from_pretrained(modelid, revision="fp16", torch_dtype=torch.float16, use_auth_token=auth_token) 
+stable_diffusion_model.to(device) 
+
+# Generate image from test 
+def generate(): 
+    """ This function generate image from a text with stable diffusion"""
+    with autocast(device): 
+        image = stable_diffusion_model(prompt.get(), guidance_scale=8.5)["sample"][0]
+    
+    # Save the generated image
+    image.save('generatedimage.png')
+    
+    # Display the generated image on the user interface
+    img = ImageTk.PhotoImage(image)
+    img_placeholder.configure(image=img) 
+
+
+trigger = ctk.CTkButton(height=40, width=120, text_font=("Arial", 15), text_color="black", fg_color="white", command=generate) 
+trigger.configure(text="Generate")
+trigger.place(x=206, y=60) 
+
+app.mainloop()

authtoken.py

@@ -0,0 +1,2 @@
+auth_token = "YOUR HUGGING FACE TOKEN HERE"
+# How to get one: https://huggingface.co/docs/hub/security-tokens

requirements.txt

@@ -0,0 +1,26 @@
+certifi==2022.9.14
+charset-normalizer==2.1.1
+colorama==0.4.5
+customtkinter==4.6.1
+darkdetect==0.7.1
+diffusers==0.3.0
+filelock==3.8.0
+huggingface-hub==0.9.1
+idna==3.4
+importlib-metadata==4.12.0
+numpy==1.23.3
+packaging==21.3
+Pillow==9.2.0
+pyparsing==3.0.9
+PyYAML==6.0
+regex==2022.9.13
+requests==2.28.1
+tk==0.1.0
+torch
+torchaudio
+torchvision
+tqdm==4.64.1
+transformers==4.22.1
+typing_extensions==4.3.0
+urllib3==1.26.12
+zipp==3.8.1