Initial test

Dockerfile

@@ -0,0 +1,14 @@
+# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

code/app.py

@@ -0,0 +1,33 @@
+from diffusers import AutoPipelineForText2Image
+import torch
+import gradio as gr
+
+pipeline_text2image = AutoPipelineForText2Image.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0", use_safetensors=True
+)
+
+def gradio_txt2img(prompt):
+    return pipeline_text2image(prompt=prompt).images[0]
+
+def api_txt2base64(prompt):
+    return gradio_txt2img(prompt).to_base64()
+
+with gr.Blocks() as demo:
+    gr.Markdown("## Stable Diffusion XL - Modded to Hell Edition")
+    gr.Markdown("## Text to Image\n\nConverts a prompt into an image.")
+    with gr.Row():
+        with gr.Column():
+            txt2img_prompt = gr.Textbox(label="Input", lines=2, max_lines=2)
+            txt2img_btn = gr.Button("Generate")
+        with gr.Column():
+            txt2img_out = gr.Image()
+    with gr.Row():
+        with gr.Column():
+            txt2b64_prompt = gr.Textbox(label="Input", lines=2, max_lines=2)
+            txt2b64_btn = gr.Button("Generate")
+        with gr.Column():
+            txt2b64_out = gr.Image()
+    txt2img_btn.click(fn=gradio_txt2img, inputs=txt2img_prompt, outputs=txt2img_out, queue=True, api_name="gradio_txt2img")
+    txt2b64_btn.click(fn=gradio_txt2img, inputs=txt2b64_prompt, outputs=txt2b64_out, queue=True, api_name="gradio_txt2b64")
+
+demo.launch()

code/requirements.txt

@@ -0,0 +1,6 @@
+gradio
+transformers
+diffusers
+accelerate
+safetensors
+omegaconf