up

README.md

@@ -1,13 +1,12 @@
 ---
-title: Vq Vs Stable Diffusion
-emoji: 👁
-colorFrom: red
-colorTo: red
+title: I Like Flan
+emoji: 🍮
+colorFrom: gray
+colorTo: purple
 sdk: gradio
-sdk_version: 3.9
+sdk_version: 3.6
 app_file: app.py
 pinned: false
-license: mit
 ---
 
 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
+import torch
+from diffusers import DiffusionPipeline
+
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
+if torch.cuda.is_available():
+    pipe_sd = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16", use_auth_token=os.getenv("HUGGING_FACE_HUB_TOKEN"))
+    pipe_vq = DiffusionPipeline.from_pretrained("microsoft/vq-diffusion-ithq", torch_dtype=torch.float16, revision="fp16")
+else:
+    pipe_sd = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=os.getenv("HUGGING_FACE_HUB_TOKEN"))
+    pipe_vq = DiffusionPipeline.from_pretrained("microsoft/vq-diffusion-ithq")
+
+examples = [
+  ["An astronaut riding a horse."],
+  ["A teddy bear playing in the water."],
+  ["A simple wedding cake with lego bride and groom topper and cake pops."],
+  ["A realistic tree using a mixture of different colored pencils."],
+  ["Muscular Santa Claus."],
+  ["A man with a pineapple head."],
+  ["Pebble tower standing on the left on the sea beach."],
+]
+
+title = "VQ Diffusion vs. Stable Diffusion 1-5"
+description = "This demo compares [VQ-Diffusion-ITHQ](https://huggingface.co/microsoft/vq-diffusion-ithq) and [Stable-Diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) for text to image generation."
+
+
+def inference(text):
+    output_sd = pipe_sd(text).images[0]
+    output_vq_diffusion = pipe_vq(text, truncation_rate=0.86).images[0]
+    return [output_vq_diffusion, output_sd]
+
+
+io = gr.Interface(
+  inference,
+  gr.Textbox(lines=3),
+  outputs=[
+    gr.Image(type="pil", label="VQ-Diffusion"),
+    gr.Image(type="pil", label="Stable Diffusion"),
+  ],
+  title=title,
+  description=description,
+  examples=examples
+)
+io.launch()

requirements.txt

@@ -0,0 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
+transformers
+diffusers[torch]