first commit

README.md

@@ -1,10 +1,10 @@
 ---
-title: Text2image
-emoji: 🔥
-colorFrom: purple
-colorTo: blue
+title: Sdxl Demo
+emoji: 📚
+colorFrom: blue
+colorTo: gray
 sdk: gradio
-sdk_version: 4.4.1
+sdk_version: 3.47.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,62 @@
+import numpy as np
+import gradio as gr
+
+import os
+from io import BytesIO
+
+import gradio as gr
+import numpy as np
+import replicate
+import requests
+from PIL import Image
+
+
+def generate(prompt: str, secret_key: str, steps: int = 25, seed: int = -1):
+    """
+    プロンプトから生成画像(PIL.Image.open)を取得
+    """
+    if secret_key == os.environ["SECRET_KEY"]:
+        output = replicate.run(
+            "stability-ai/sdxl:2b017d9b67edd2ee1401238df49d75da53c523f36e363881e057f5dc3ed3c5b2",
+            input={"prompt": prompt,
+                   "seed": seed if seed != -1 else np.random.randint(1, 1001),
+                   "num_inference_steps": steps},
+        )
+        # リンク取得
+        png_link = output[0]
+        # PNGファイルをリンクから取得
+        response = requests.get(png_link)
+        # イメージをメモリ上に開く
+        img = Image.open(BytesIO(response.content))
+        return img
+
+
+
+default_steps = 25
+examples = [
+    # ["An astronaut riding a rainbow unicorn, cinematic, dramatic", ""],
+    # ["A robot painted as graffiti on a brick wall. a sidewalk is in front of the wall, and grass is growing out of cracks in the concrete.", ""],
+    # ["Panda mad scientist mixing sparkling chemicals, artstation.", ""],
+    ["An astronaut riding a rainbow unicorn, cinematic, dramatic"],
+    ["photo of a rhino dressed suit and tie sitting at a table in a bar with a bar stools, award winning photography."],
+    ["a giant monster hybrid of dragon and spider, in dark dense foggy forest"],
+    ["a man in a space suit playing a piano, highly detailed illustration, full color illustration, very detailed illustration"],
+    ]
+
+with gr.Blocks(title="Stable Diffusion XL (SDXL 1.0)") as demo:
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            gr_prompt = gr.Textbox(label='プロンプト')
+            gr_password = gr.Textbox(label='パスワード')
+            gr_generate_button = gr.Button("生成")
+            with gr.Accordion("advanced settings", open=False):
+                gr_steps = gr.Number(label='steps', value=default_steps)
+                gr_seed = gr.Number(label='seed', value=-1)
+        with gr.Column(scale=1, min_width=600):
+            gr_image = gr.Image()
+        # examples=examples
+        gr_generate_button.click(generate, inputs=[gr_prompt, gr_password, gr_steps, gr_seed], outputs=[gr_image])
+    with gr.Row():
+        gr.Examples(examples, inputs=[gr_prompt])
+
+demo.launch()

gitattributes

@@ -0,0 +1,35 @@
+*.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
+*.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

requirements.txt

@@ -0,0 +1,3 @@
+replicate==0.10.0
+gradio==3.39.0
+numpy==1.22.4