Update app.py

app.py

@@ -34,35 +34,14 @@ positive_prompt_str = "Full HD, 4K, high quality, high resolution"
 examples_path = root_path / '__assets__/demo/examples'
 example_inputs = [
     [f'{examples_path}/images_1024/a40.jpg', f'{examples_path}/images_2048/a40.jpg', 'medieval castle'],
-    [f'{examples_path}/images_1024/a4.jpg', f'{examples_path}/images_2048/a4.jpg', 'parrot'],
-    [f'{examples_path}/images_1024/a65.jpg', f'{examples_path}/images_2048/a65.jpg', 'hoodie'],
-    [f'{examples_path}/images_1024/a54.jpg', f'{examples_path}/images_2048/a54.jpg', 'salad'],
-    [f'{examples_path}/images_1024/a51.jpg', f'{examples_path}/images_2048/a51.jpg', 'space helmet'],
-    [f'{examples_path}/images_1024/a46.jpg', f'{examples_path}/images_2048/a46.jpg', 'stack of books'],
-    [f'{examples_path}/images_1024/a19.jpg', f'{examples_path}/images_2048/a19.jpg', 'antique greek vase'],
-    [f'{examples_path}/images_1024/a2.jpg', f'{examples_path}/images_2048/a2.jpg', 'sunglasses'],
 ]
 
 thumbnails = [
     'https://lh3.googleusercontent.com/pw/ABLVV87bkFc_SRKrbXuk5BTp18dETNm18MLbjoJo6JvwbIkYtjZXrjU_H1dCJIP799OJjHTZmo19mYVyMCC1RLmwqzoZrgwQzfB-SCtxLa83IbXBQ23xzmKoZgsRlPztxNJD6gmXzFyatdLRzDxHIusBQLUz=w3580-h1150-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV85RWtrpTf1tMp2p3q37eg5DlFp5znifALK_JTjvxJua8UYMjytVoEy2GUW2cLXgBvQyYKg7GvrWXQ5hkdAsyih5Rf4rFnDq-JoiQYhVZHStCZLKxmeAlQna5ZwMPVTKG1TK63DH_OdK58gvSjWtF2ww=w3580-h1152-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV84dkaU6SQs9fyDjajpk1X9JkYp_zQBEnPVL67oi11_05U6-Ys5ydQpuny8GBQCMyVbFKxJ5unn9w__gmP9K0cKQ4_IVoT7Hvfmya71klDqSI7vu9Iy_5P2Il5-0giJFpumtffBA3kryn1xtJdR4vSA0=w2924-h1858-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV853ZyjvS4LvcPpVMY9BWz-232omt3-hgRiGcky_3ojE6WLKgtsrftsg1jSrUm2ccT_UOa279CulZy6fdnH_Xg1SunyRBxaRjOK0uxAkUFwb60rR1S4hI2MmhLV7KCi3tw1A-oiGi0f9JINyade-322A=w2622-h1858-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV86AJGUVGjb0i6CPg8zlJlWObNY0xdOzM1x5Bq9gKhP-ZWre5aaexRJDxQUO2gmJtRIyohD88FJDG_aVX2G5M0QOyGRWlZmx7tOVXLh-Kbesobxo9MfD-wqk9Ts9O8NUGtIwkWzo9SEs2opKdu83gB9F=w2528-h1858-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV87MplTciS7z-4i-eY3B3L0YhaK8UEQ3pTQD6W6uYVGR4hPD9u1WGEGyfg5ddqU-Bx2BrKskDhwxzF746cRhgFU5aPtbYA_-O7KfqXe9IsMxYCgUKxEHBm2ncqy64V-w-N8XOFgUMkAQqcuuNZ8Xapqp=w3580-h1186-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV877Esi6l2Kuw3akH5QBlmDAbWydZDZEEJqlZ_N-X7g33NQZU8nv_UKdAVETS7q23byTuldIAhW-q99zCycFB8Yfc-5e_WPNIM9icU0p3gd6DUVZR233ZNUtLca384MYGIhMGud9Y_Xed1I3PpiMhrpG=w2846-h1858-s-no-gm',
-    'https://lh3.googleusercontent.com/pw/ABLVV85hMQbSB6fCokdyut4ke7xTUqjERhuYygnj7T8IIA1k48e9GkaowDywPZzi5QJzZfj7wU3bgBHzjxop19qK1zOi5XDrjfXkn5bwj4MxicHa3TG-Rc-V-c1uyZVUyviyUlkGZ62FxuVROw2x0aGJIcr0=w3580-h1382-s-no-gm'
 ]
 
 example_previews = [
     [thumbnails[0], 'Prompt: medieval castle'],
-    [thumbnails[1], 'Prompt: parrot'],
-    [thumbnails[2], 'Prompt: hoodie'],
-    [thumbnails[3], 'Prompt: salad'],
-    [thumbnails[4], 'Prompt: space helmet'],
-    [thumbnails[5], 'Prompt: stack of books'],
-    [thumbnails[6], 'Prompt: antique greek vase'],
-    [thumbnails[7], 'Prompt: sunglasses'],
 ]
 
 # Load models
@@ -244,34 +223,10 @@ def upscale_run(
 with gr.Blocks(css=demo_path / 'style.css') as demo:
     gr.HTML(
         """
-        <div style="text-align: center; max-width: 1200px; margin: 20px auto;">
-        <h1 style="font-weight: 900; font-size: 3rem; margin-bottom: 0.5rem">
-            🧑‍🎨 HD-Painter Demo
-        </h1>
-        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
-        Hayk Manukyan<sup>1*</sup>, Andranik Sargsyan<sup>1*</sup>, Barsegh Atanyan<sup>1</sup>, Zhangyang Wang<sup>1,2</sup>, Shant Navasardyan<sup>1</sup>
-        and <a href="https://www.humphreyshi.com/home">Humphrey Shi</a><sup>1,3</sup>
-        </h2>
-        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
-        <sup>1</sup>Picsart AI Resarch (PAIR), <sup>2</sup>UT Austin, <sup>3</sup>Georgia Tech
-        </h2>
-        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
-        [<a href="https://arxiv.org/abs/2312.14091" style="color:blue;">arXiv</a>] 
-        [<a href="https://github.com/Picsart-AI-Research/HD-Painter" style="color:blue;">GitHub</a>]
-        </h2>
-        <h2 style="font-weight: 450; font-size: 1rem; margin: 0.7rem auto; max-width: 1000px">
-        <b>HD-Painter</b> enables prompt-faithfull and high resolution (up to 2k) image inpainting upon any diffusion-based image inpainting method.
-        </h2>
-        </div>
         """)
 
     if on_huggingspace:
-        gr.HTML("""
-        <p>For faster inference without waiting in queue, you may duplicate the space and upgrade to the suggested GPU in settings.
-        <br/>
-        <a href="https://huggingface.co/spaces/PAIR/HD-Painter?duplicate=true">
-        <img style="margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
-        </p>""")
+        gr.HTML("""""")
 
     with open(demo_path / 'script.js', 'r') as f:
         js_str = f.read()
@@ -325,9 +280,9 @@ with gr.Blocks(css=demo_path / 'style.css') as demo:
                     elem_id='outputgallery'
                 )
             with gr.Row():
-                upscale_btn = gr.Button("Send to Inpainting-Specialized Super-Resolution (x4)", scale = 1)
+                upscale_btn = gr.Button("Send to Upscaler (x4)", scale = 1)
             with gr.Row():
-                use_sam_mask = gr.Checkbox(value = False, label = "Use SAM mask for background preservation (for SR only, experimental feature)")
+                use_sam_mask = gr.Checkbox(value = False, label = "Use SAM mask for background preservation")
             with gr.Row():
                 hires_image = gr.Image(label = "Hi-res Image")