Upload folder using huggingface_hub

README.md

@@ -5,7 +5,7 @@ emoji: 🔥
 colorFrom: indigo
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.40.1
+sdk_version: 3.41.2
 app_file: run.py
 pinned: false
 hf_oauth: true

requirements.txt

@@ -1,4 +1,4 @@
-https://gradio-main-build.s3.amazonaws.com/2e720c0c47ea94923ab689f928308157f2c30b2d/gradio-3.40.1-py3-none-any.whl
+https://gradio-main-build.s3.amazonaws.com/d76d50112328711b1a692b80c1e88a085b15b301/gradio-3.41.2-py3-none-any.whl
 diffusers
 transformers
 nvidia-ml-py3

run.ipynb

@@ -1 +1 @@
-{"cells": [{"cell_type": "markdown", "id": 302934307671667531413257853548643485645, "metadata": {}, "source": ["# Gradio Demo: stable-diffusion\n", "### Note: This is a simplified version of the code needed to create the Stable Diffusion demo. See full code here: https://hf.co/spaces/stabilityai/stable-diffusion/tree/main\n", "        "]}, {"cell_type": "code", "execution_count": null, "id": 272996653310673477252411125948039410165, "metadata": {}, "outputs": [], "source": ["!pip install -q gradio diffusers transformers nvidia-ml-py3 ftfy torch"]}, {"cell_type": "code", "execution_count": null, "id": 288918539441861185822528903084949547379, "metadata": {}, "outputs": [], "source": ["import gradio as gr\n", "import torch\n", "from diffusers import StableDiffusionPipeline\n", "from PIL import Image  \n", "import os\n", "\n", "auth_token = os.getenv(\"auth_token\")\n", "model_id = \"CompVis/stable-diffusion-v1-4\"\n", "device = \"cpu\"\n", "pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token, revision=\"fp16\", torch_dtype=torch.float16)\n", "pipe = pipe.to(device)\n", "\n", "def infer(prompt, samples, steps, scale, seed):        \n", "    generator = torch.Generator(device=device).manual_seed(seed)\n", "    images_list = pipe(\n", "        [prompt] * samples,\n", "        num_inference_steps=steps,\n", "        guidance_scale=scale,\n", "        generator=generator,\n", "    )\n", "    images = []\n", "    safe_image = Image.open(r\"unsafe.png\")\n", "    for i, image in enumerate(images_list[\"sample\"]):\n", "        if(images_list[\"nsfw_content_detected\"][i]):\n", "            images.append(safe_image)\n", "        else:\n", "            images.append(image)\n", "    return images\n", "    \n", "\n", "\n", "block = gr.Blocks()\n", "\n", "with block:\n", "    with gr.Group():\n", "        with gr.Box():\n", "            with gr.Row().style(mobile_collapse=False, equal_height=True):\n", "                text = gr.Textbox(\n", "                    label=\"Enter your prompt\",\n", "                    show_label=False,\n", "                    max_lines=1,\n", "                    placeholder=\"Enter your prompt\",\n", "                ).style(\n", "                    border=(True, False, True, True),\n", "                    rounded=(True, False, False, True),\n", "                    container=False,\n", "                )\n", "                btn = gr.Button(\"Generate image\").style(\n", "                    margin=False,\n", "                    rounded=(False, True, True, False),\n", "                )\n", "        gallery = gr.Gallery(\n", "            label=\"Generated images\", show_label=False, elem_id=\"gallery\"\n", "        ).style(grid=[2], height=\"auto\")\n", "\n", "        advanced_button = gr.Button(\"Advanced options\", elem_id=\"advanced-btn\")\n", "\n", "        with gr.Row(elem_id=\"advanced-options\"):\n", "            samples = gr.Slider(label=\"Images\", minimum=1, maximum=4, value=4, step=1)\n", "            steps = gr.Slider(label=\"Steps\", minimum=1, maximum=50, value=45, step=1)\n", "            scale = gr.Slider(\n", "                label=\"Guidance Scale\", minimum=0, maximum=50, value=7.5, step=0.1\n", "            )\n", "            seed = gr.Slider(\n", "                label=\"Seed\",\n", "                minimum=0,\n", "                maximum=2147483647,\n", "                step=1,\n", "                randomize=True,\n", "            )\n", "        text.submit(infer, inputs=[text, samples, steps, scale, seed], outputs=gallery)\n", "        btn.click(infer, inputs=[text, samples, steps, scale, seed], outputs=gallery)\n", "        advanced_button.click(\n", "            None,\n", "            [],\n", "            text,\n", "        )\n", "        \n", "block.launch()"]}], "metadata": {}, "nbformat": 4, "nbformat_minor": 5}
+{"cells": [{"cell_type": "markdown", "id": 302934307671667531413257853548643485645, "metadata": {}, "source": ["# Gradio Demo: stable-diffusion\n", "### Note: This is a simplified version of the code needed to create the Stable Diffusion demo. See full code here: https://hf.co/spaces/stabilityai/stable-diffusion/tree/main\n", "        "]}, {"cell_type": "code", "execution_count": null, "id": 272996653310673477252411125948039410165, "metadata": {}, "outputs": [], "source": ["!pip install -q gradio diffusers transformers nvidia-ml-py3 ftfy torch"]}, {"cell_type": "code", "execution_count": null, "id": 288918539441861185822528903084949547379, "metadata": {}, "outputs": [], "source": ["import gradio as gr\n", "import torch\n", "from diffusers import StableDiffusionPipeline\n", "from PIL import Image\n", "import os\n", "\n", "auth_token = os.getenv(\"auth_token\")\n", "model_id = \"CompVis/stable-diffusion-v1-4\"\n", "device = \"cpu\"\n", "pipe = StableDiffusionPipeline.from_pretrained(\n", "    model_id, use_auth_token=auth_token, revision=\"fp16\", torch_dtype=torch.float16\n", ")\n", "pipe = pipe.to(device)\n", "\n", "\n", "def infer(prompt, samples, steps, scale, seed):\n", "    generator = torch.Generator(device=device).manual_seed(seed)\n", "    images_list = pipe(\n", "        [prompt] * samples,\n", "        num_inference_steps=steps,\n", "        guidance_scale=scale,\n", "        generator=generator,\n", "    )\n", "    images = []\n", "    safe_image = Image.open(r\"unsafe.png\")\n", "    for i, image in enumerate(images_list[\"sample\"]):\n", "        if images_list[\"nsfw_content_detected\"][i]:\n", "            images.append(safe_image)\n", "        else:\n", "            images.append(image)\n", "    return images\n", "\n", "\n", "block = gr.Blocks()\n", "\n", "with block:\n", "    with gr.Group():\n", "        with gr.Row():\n", "            text = gr.Textbox(\n", "                label=\"Enter your prompt\",\n", "                max_lines=1,\n", "                placeholder=\"Enter your prompt\",\n", "                container=False,\n", "            )\n", "            btn = gr.Button(\"Generate image\")\n", "        gallery = gr.Gallery(\n", "            label=\"Generated images\",\n", "            show_label=False,\n", "            elem_id=\"gallery\",\n", "            columns=[2],\n", "            height=\"auto\",\n", "        )\n", "\n", "        advanced_button = gr.Button(\"Advanced options\", elem_id=\"advanced-btn\")\n", "\n", "        with gr.Row(elem_id=\"advanced-options\"):\n", "            samples = gr.Slider(label=\"Images\", minimum=1, maximum=4, value=4, step=1)\n", "            steps = gr.Slider(label=\"Steps\", minimum=1, maximum=50, value=45, step=1)\n", "            scale = gr.Slider(\n", "                label=\"Guidance Scale\", minimum=0, maximum=50, value=7.5, step=0.1\n", "            )\n", "            seed = gr.Slider(\n", "                label=\"Seed\",\n", "                minimum=0,\n", "                maximum=2147483647,\n", "                step=1,\n", "                randomize=True,\n", "            )\n", "        text.submit(infer, inputs=[text, samples, steps, scale, seed], outputs=gallery)\n", "        btn.click(infer, inputs=[text, samples, steps, scale, seed], outputs=gallery)\n", "        advanced_button.click(\n", "            None,\n", "            [],\n", "            text,\n", "        )\n", "\n", "block.launch()\n"]}], "metadata": {}, "nbformat": 4, "nbformat_minor": 5}

run.py

@@ -1,16 +1,19 @@
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
-from PIL import Image  
+from PIL import Image
 import os
 
 auth_token = os.getenv("auth_token")
 model_id = "CompVis/stable-diffusion-v1-4"
 device = "cpu"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token, revision="fp16", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id, use_auth_token=auth_token, revision="fp16", torch_dtype=torch.float16
+)
 pipe = pipe.to(device)
 
-def infer(prompt, samples, steps, scale, seed):        
+
+def infer(prompt, samples, steps, scale, seed):
     generator = torch.Generator(device=device).manual_seed(seed)
     images_list = pipe(
         [prompt] * samples,
@@ -21,37 +24,32 @@ def infer(prompt, samples, steps, scale, seed):
     images = []
     safe_image = Image.open(r"unsafe.png")
     for i, image in enumerate(images_list["sample"]):
-        if(images_list["nsfw_content_detected"][i]):
+        if images_list["nsfw_content_detected"][i]:
             images.append(safe_image)
         else:
             images.append(image)
     return images
-    
 
 
 block = gr.Blocks()
 
 with block:
     with gr.Group():
-        with gr.Box():
-            with gr.Row().style(mobile_collapse=False, equal_height=True):
-                text = gr.Textbox(
-                    label="Enter your prompt",
-                    show_label=False,
-                    max_lines=1,
-                    placeholder="Enter your prompt",
-                ).style(
-                    border=(True, False, True, True),
-                    rounded=(True, False, False, True),
-                    container=False,
-                )
-                btn = gr.Button("Generate image").style(
-                    margin=False,
-                    rounded=(False, True, True, False),
-                )
+        with gr.Row():
+            text = gr.Textbox(
+                label="Enter your prompt",
+                max_lines=1,
+                placeholder="Enter your prompt",
+                container=False,
+            )
+            btn = gr.Button("Generate image")
         gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+            label="Generated images",
+            show_label=False,
+            elem_id="gallery",
+            columns=[2],
+            height="auto",
+        )
 
         advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
 
@@ -75,5 +73,5 @@ with block:
             [],
             text,
         )
-        
-block.launch()
+
+block.launch()