Update app.py

app.py

@@ -38,7 +38,6 @@ def run(image, item_name):
     prompt = item_name + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
     image = image.convert("RGB")
     image_t = TF.to_tensor(image) > 0.5
-    image_pil = TF.to_pil_image(image_t.to(torch.float32))
     with torch.no_grad():
         c_t = image_t.unsqueeze(0).cuda().float()
         torch.manual_seed(42)
@@ -46,9 +45,7 @@ def run(image, item_name):
         noise = torch.randn((1,4,H//8, W//8), device=c_t.device)
         output_image = model(c_t, prompt, deterministic=False, r=0.4, noise_map=noise)
     output_pil = TF.to_pil_image(output_image[0].cpu()*0.5+0.5)
-    input_sketch_uri = pil_image_to_data_uri(Image.fromarray(255-np.array(image)))
-    output_image_uri = pil_image_to_data_uri(output_pil)
-    return output_pil, gr.update(link=input_sketch_uri), gr.update(link=output_image_uri)
+    return output_pil
 
 
 def update_canvas(use_line, use_eraser):
@@ -164,10 +161,10 @@ async () => {
 
 with gr.Blocks(css="style.css") as demo:
     # these are hidden buttons that are used to trigger the canvas changes
-    line = gr.Checkbox(label="line", value=False, elem_id="cb-line")
-    eraser = gr.Checkbox(label="eraser", value=False, elem_id="cb-eraser")
-    with gr.Row(elem_id="header_block"):
-        gr.HTML("""<img src="file=assets/title.png" alt="title" /><img src="file=assets/logos.png" alt="title" />""")
+    gr.HTML("""<div id="header_block">
+      <img src="file=assets/title.png" alt="title" />
+      <img src="file=assets/logos.png" alt="title" />
+    </div>""")
     with gr.Column(elem_id="main_block"):
         with gr.Row(elem_id="board_row"):
             with gr.Group(elem_id="input_image_container", elem_classes="image_container" ):
@@ -183,8 +180,9 @@ with gr.Blocks(css="style.css") as demo:
                 result = gr.Image(label="Result", height=440, width=440, elem_id="output_image", show_label=False, show_download_button=True)
         with gr.Row(elem_id="radio_row"):
             item = gr.Radio(choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME)
-    with gr.Row(elem_id="footer_block"):
-        gr.HTML("""<button id="fullscreen" onclick="document.body.querySelector('#main_row').requestFullscreen();" class="fullscreen" title="fullscreen"></button>""")
+    gr.HTML("""<div id="footer_block">
+        <button id="fullscreen" onclick="document.body.querySelector('#main_row').requestFullscreen();" class="fullscreen" title="fullscreen"></button>
+        </div>""")
 
     demo.load(None,None,None,_js=scripts)
     inputs = [image, item]