Upload 4 files

README.md

@@ -1,12 +1,13 @@
 ---
-title: Testimg
-emoji: ⚡
-colorFrom: pink
-colorTo: pink
+title: SD Txt2img
+emoji: 😻
+colorFrom: purple
+colorTo: gray
 sdk: gradio
-sdk_version: 3.35.2
+sdk_version: 3.29.0
 app_file: app.py
 pinned: false
+license: creativeml-openrail-m
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,65 @@
+import gradio as gr
+
+def generateImage(prompt, n_prompt, modelName):
+
+    return models[modelName].process(prompt, n_prompt)
+
+
+
+
+def create_demo():
+
+    with gr.Blocks() as demo:
+        with gr.Column():
+            prompt = gr.Textbox(label='Prompt')
+            n_prompt = gr.Textbox(
+                label='Negative Prompt',
+                value=
+                'ugly, disfigured, deformed'
+            )
+
+            modelName = gr.Dropdown(choices = list(models.keys()), 
+                                        label = "Model", 
+                                        value=list(models.keys())[0])
+
+            run_button = gr.Button('Run')
+        gr.Markdown("### [Stable Diffusion Art](https://stable-diffusion-art.com/) -- tutorials and resources. Read [Model license](https://huggingface.co/spaces/CompVis/stable-diffusion-license).")
+        result = gr.Gallery(label='Output',
+                            show_label=False,
+                            elem_id='gallery').style(columns=1, rows=1, preview=True)        
+
+
+        inputs = [
+            prompt,
+            n_prompt,
+            modelName,
+        ]
+
+        prompt.submit(
+            fn=generateImage,
+            inputs=inputs,
+            outputs=result
+        )
+        n_prompt.submit(
+            fn=generateImage,
+            inputs=inputs,
+            outputs=result
+        )
+
+        run_button.click(
+            fn=generateImage,
+            inputs=inputs,
+            outputs=result
+        )
+    return demo
+
+
+if __name__ == '__main__':
+    from model import Model
+    models = {
+        "Stable Diffusion v1.5": Model("runwayml/stable-diffusion-v1-5"),
+        "Realistic Vision v2.0": Model("SG161222/Realistic_Vision_V2.0"),
+        "Anything v3.0": Model("Linaqruf/anything-v3.0")
+              }
+    demo = create_demo()
+    demo.queue().launch()

model.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+from diffusers import DPMSolverMultistepScheduler
+import torch
+import PIL.Image
+import numpy as np
+import datetime
+
+# # Check environment
+# print(f"Is CUDA available: {torch.cuda.is_available()}")
+# # True
+# print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# # Tesla T4
+
+
+device = "cpu"
+
+
+
+class Model:
+    def __init__(self, modelID):
+        #modelID = "runwayml/stable-diffusion-v1-5"
+
+        self.modelID = modelID
+        self.pipe = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
+        self.pipe = self.pipe.to(device)
+        self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
+        self.pipe.enable_xformers_memory_efficient_attention()
+
+        #self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
+        #prompt = "a photo of an astronaut riding a horse on mars"
+        #n_prompt = "deformed, disfigured"
+
+    def process(self, 
+                prompt: str, 
+                negative_prompt: str,
+                guidance_scale:int = 7,
+                num_images:int = 1,
+                num_steps:int = 20,
+                ):
+        seed = np.random.randint(0, np.iinfo(np.int32).max)
+        generator = torch.Generator(device).manual_seed(seed)
+        now = datetime.datetime.now()
+        print(now)
+        print(self.modelID)
+        print(prompt)
+        print(negative_prompt)
+        with torch.inference_mode():
+            images = self.pipe(prompt=prompt,
+                         negative_prompt=negative_prompt,
+                         guidance_scale=guidance_scale,
+                         num_images_per_prompt=num_images,
+                         num_inference_steps=num_steps,
+                         generator=generator).images
+
+        return images
+
+
+
+# image = pipeline(prompt=prompt,
+#                  negative_prompt = n_prompt, 
+#                  num_inference_steps = 2,
+#                  guidance_scale = 7).images
+
+

requirements.txt

@@ -0,0 +1,9 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
+accelerate==0.18.0
+diffusers==0.16.0
+gradio==3.30.0
+safetensors==0.3.0
+torchvision==0.15.1
+transformers==4.28.1
+xformers==0.0.18