Styling and Performance

app.py

@@ -19,6 +19,7 @@ auth_token = os.environ.get("API_TOKEN") or True
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", dtype=torch.float16, revision="fp16", use_auth_token=auth_token)
+pipe.to(device)
 
 transform = transforms.Compose([
       transforms.ToTensor(),
@@ -35,10 +36,9 @@ def read_content(file_path: str) -> str:
     return content
 
 def predict(dict, prompt=""):
-    with autocast("cuda"):
-        init_image = dict["image"].convert("RGB").resize((512, 512))
-        mask = dict["mask"].convert("RGB").resize((512, 512))
-        output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
+    init_image = dict["image"].convert("RGB").resize((512, 512))
+    mask = dict["mask"].convert("RGB").resize((512, 512))
+    output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
     return output.images[0]
 
 
@@ -65,7 +65,7 @@ with image_blocks as demo:
             image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload").style(height=400)
 
             with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-                prompt = gr.Textbox(label = 'Your prompt (what you want to replace)')
+                prompt = gr.Textbox(label = 'Your prompt (what you want in place of what is erased)')
                 btn = gr.Button("Generate image").style(
                     margin=False,
                     rounded=(False, True, True, False),

header.html

@@ -1,38 +1,23 @@
 <div style="text-align: center; max-width: 650px; margin: 0 auto;">
-    <div
-      style="
+  <div style="
         display: inline-flex;
         gap: 0.8rem;
         font-size: 1.75rem;
-        margin-bottom: 10px;
-        margin-left: 220px;
-        justify-content: center;
-      "
-    >
-    <a href="https://runwayml.com/"><img src="https://aeiljuispo.cloudimg.io/v7/https://s3.amazonaws.com/moonup/production/uploads/1665970599545-634cb15a4abe8405758d2e7e.jpeg" alt="RunwayML" width="30%"></a>
-    </div> 
-    <div
-      style="
-        display: inline-flex;
-        align-items: center;
-        gap: 0.8rem;
-        font-size: 1.75rem;
-        margin-bottom: 10px;
         justify-content: center;
       ">
-    <a href="https://github.com/PaddlePaddle/PaddleNLP/tree/develop/model_zoo/ernie-layout"><h1 style="font-weight: 900; align-items: center; margin-bottom: 7px;">
+    <img
+      src="https://aeiljuispo.cloudimg.io/v7/https://s3.amazonaws.com/moonup/production/uploads/1665970599545-634cb15a4abe8405758d2e7e.jpeg"
+      alt="RunwayML" width="64px">
+    <h1 style="font-weight: 900; align-items: center; margin-bottom: 7px; margin-top: 12px;">
       RunwayML Stable Diffusion Inpainting 🎨
-    </h1></a>
-    </div>
-    <div>
-    <p style="font-weight: 900; align-items: center; margin-bottom: 7px;">
-        Stable Diffusion Inpainting by RunwayML, add a mask and text prompt for what you want to replace <br>For faster generation you can try
-        <a
-          href="https://app.runwayml.com/video-tools/teams/akhaliq/ai-tools/erase-and-replace"
-          style="text-decoration: underline;"
-          target="_blank"
-          >erase and replace tool on Runway</a
-        >
-      </p>
-    </div> 
-  </div>
+    </h1>
+  </div>
+  <div>
+    <p style="align-items: center; margin-bottom: 7px;">
+      Stable Diffusion Inpainting, add a mask and text prompt for what you want to replace <br>For faster
+      generation you can try
+      <a href="https://app.runwayml.com/video-tools/teams/akhaliq/ai-tools/erase-and-replace"
+        style="text-decoration: underline;" target="_blank">erase and replace tool on Runway</a>
+    </p>
+  </div>
+</div>