Overhaul style

app.py

@@ -1,15 +1,66 @@
+# !pip install -Uq diffusers transformers
+# !pip install -Uq gradio
+# !pip install -Uq accelerate
+
 import gradio
 from diffusers import StableDiffusionPipeline as pipeline
+from accelerate import init_empty_weights
 import torch
 
+from PIL import Image
+import matplotlib.pyplot as plot
+
 my_token = "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"
-pipe = pipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
+
+with init_empty_weights():
+  pipe = pipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
+
 
 def image_prompt(thing):
   prompt = "A children's drawing of a " + thing
   return pipe(prompt=prompt, height=512, width=664).images[0]
 
-with gradio.Blocks() as demo:
+with gradio.Blocks(css="""
+  #go-button {
+    background-color: white;
+    border-radius: 0;
+    border: none;
+    font-family: serif;
+    background-image: none;
+    font-weight: 100;
+    width: fit-content;
+    display: block;
+    margin-left: auto;
+    margin-right: auto;
+    text-decoration: underline;
+    box-shadow: none;
+  }
+  .rounded-lg {
+    border: none;
+  }
+  .gr-box {
+    border-radius: 0;
+    border: 1px solid black;
+  }
+  .text-gray-500 {
+    color: black;
+    font-family: serif;
+    font-size: 15px;
+  }
+  .border-gray-200 {
+    border: 1px solid black;
+  }
+  .bg-gray-200 {
+    background-color: white;
+    --tw-bg-opacity: 0;
+  }
+  footer {
+    display: none;
+  }
+  footer {
+    opacity: 0;
+  }
+""") as demo:
   animal = gradio.Textbox(label="A kid's drawing of a...")
   output = gradio.Image(label="Behold...")
   go_button = gradio.Button("Draw it!")