Duplicate from AlStable/stable-diffusion-2-1

Co-authored-by: clifford padraig <[email protected]>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Stable Diffusion 2 1
+emoji: 🤗
+colorFrom: indigo
+colorTo: purple
+sdk: gradio
+sdk_version: 3.15.0
+app_file: app.py
+pinned: false
+duplicated_from: AlStable/stable-diffusion-2-1
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,47 @@
+import os
+import gradio as gr
+
+API_KEY=os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
+article = """---
+This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""
+
+class Model:
+    def __init__(self, name, path="", prefix=""):
+        self.name = name
+        self.path = path
+        self.prefix = prefix
+        self.pipe_t2i = None
+        self.pipe_i2i = None
+
+models = [
+   Model("Marvel","models/ItsJayQz/Marvel_WhatIf_Diffusion", "whatif style"), 
+   Model("Cyberpunk Anime Diffusion", "models/DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style"),
+   Model("Guan Yu Diffusion", "models/DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu style"),
+   Model("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
+   Model("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
+   Model("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 default style")
+]
+
+custom_model = "models/stabilityai/stable-diffusion-2-1"
+
+def selectModel(message):
+    message = message.lower()
+    for i in range(len(models)):
+        if message.find(models[i].prefix)!=-1:
+            c_model=models[i].path
+            return c_model
+        else:
+            c_model=models[i].path
+    return c_model
+
+sandbox = gr.Interface.load(
+    fn= selectModel,
+    name= custom_model, 
+    title="""AlStable sandbox""",
+    inputs = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt", elem_id="input-prompt"), 
+    description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">AlStable</a> Stable Diffusion model.""",
+    article=article,
+    api_key=API_KEY
+)
+
+sandbox.queue(concurrency_count=20).launch()