Update gradio_demo.py

gradio_demo.py

@@ -327,7 +327,6 @@ with gr.Blocks(title="SUPIR") as interface:
     input_image = gr.Image(label="Input", show_label=True, type="numpy", height=600, elem_id="image-input")
     with gr.Group():
         prompt = gr.Textbox(label="Image description for LlaVa", value="", placeholder="A person, walking, in a town, Summer, photorealistic", lines=3, visible=False)
-        upscale = gr.Radio([["x1", 1], ["x2", 2], ["x3", 3], ["x4", 4], ["x5", 5], ["x6", 6], ["x7", 7], ["x8", 8]], label="Upscale factor", info="Resolution x1 to x8", value=2, interactive=True)
         a_prompt = gr.Textbox(label="Image description",
                               info="Help the AI understand what the image represents; describe as much as possible",
                               value='Cinematic, High Contrast, highly detailed, taken using a Canon EOS R '
@@ -336,6 +335,7 @@ with gr.Blocks(title="SUPIR") as interface:
                                     'hyper sharpness, perfect without deformations.',
                               lines=3)
         a_prompt_hint = gr.HTML("You can use a <a href='"'https://huggingface.co/spaces/MaziyarPanahi/llava-llama-3-8b'"'>LlaVa space</a> to auto-generate the description of your image.")
+        upscale = gr.Radio([["x1", 1], ["x2", 2], ["x3", 3], ["x4", 4], ["x5", 5], ["x6", 6], ["x7", 7], ["x8", 8]], label="Upscale factor", info="Resolution x1 to x8", value=2, interactive=True)
         output_format = gr.Radio([["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"]], label="Image format for result", info="File extention", value="png", interactive=True)
 
     with gr.Accordion("Pre-denoising (optional)", open=False):
@@ -409,9 +409,110 @@ with gr.Blocks(title="SUPIR") as interface:
         fb_score = gr.Slider(label="Feedback Score", minimum=1, maximum=5, value=3, step=1, interactive=True)
         fb_text = gr.Textbox(label="Feedback Text", value="", placeholder='Please enter your feedback here.')
         submit_button = gr.Button(value="Submit Feedback")
+        event_id = gr.Textbox(label="Event ID", value="", visible=False)
+
+    gr.Examples(
+        fn = stage2_process,
+	    inputs = [
+            input_image,
+            denoise_image,
+            prompt,
+            a_prompt,
+            n_prompt,
+            num_samples,
+            min_size,
+            downscale,
+            upscale,
+            edm_steps,
+            s_stage1,
+            s_stage2,
+            s_cfg,
+            randomize_seed,
+            seed,
+            s_churn,
+            s_noise,
+            color_fix_type,
+            diff_dtype,
+            ae_dtype,
+            gamma_correction,
+            linear_CFG,
+            linear_s_stage2,
+            spt_linear_CFG,
+            spt_linear_s_stage2,
+            model_select,
+            output_format
+        ],
+	    outputs = [
+            result_slider,
+            result_gallery,
+            restore_information,
+            event_id
+        ],
+        examples = [
+                [
+                    "./Examples/Example1.png",
+                    None,
+                    None,
+                    "Group of people, walking, happy, in the street, photorealistic, 8k, extremely detailled, Cinematic, High Contrast, highly detailed, taken using a Canon EOS R camera, hyper detailed photo - realistic maximum detail, 32k, Color Grading, ultra HD, extreme meticulous detailing, skin pore detailing, hyper sharpness, perfect without deformations.",
+                    "painting, oil painting, illustration, drawing, art, sketch, anime, cartoon, CG Style, 3D render, unreal engine, blurring, bokeh, ugly, dirty, messy, worst quality, low quality, frames, watermark, signature, jpeg artifacts, deformed, lowres, over-smooth",
+                    1,
+                    1024,
+                    1,
+                    1,
+                    200,
+                    -1,
+                    1,
+                    7.5,
+                    False,
+                    42,
+                    5,
+                    1.003,
+                    "Wavelet",
+                    "fp16",
+                    "bf16",
+                    1.0,
+                    True,
+                    4,
+                    False,
+                    0.,
+                    "v0-Q",
+                    "png"
+                ],
+                [
+                    "./Examples/Example2.jpeg",
+                    None,
+                    None,
+                    "The head of a tabby cat, in a house, photorealistic, 8k, extremely detailled, Cinematic, High Contrast, highly detailed, taken using a Canon EOS R camera, hyper detailed photo - realistic maximum detail, 32k, Color Grading, ultra HD, extreme meticulous detailing, skin pore detailing, hyper sharpness, perfect without deformations.",
+                    "painting, oil painting, illustration, drawing, art, sketch, anime, cartoon, CG Style, 3D render, unreal engine, blurring, bokeh, ugly, dirty, messy, worst quality, low quality, frames, watermark, signature, jpeg artifacts, deformed, lowres, over-smooth",
+                    1,
+                    1024,
+                    1,
+                    1,
+                    200,
+                    -1,
+                    1,
+                    7.5,
+                    False,
+                    42,
+                    5,
+                    1.003,
+                    "Wavelet",
+                    "fp16",
+                    "bf16",
+                    1.0,
+                    True,
+                    4,
+                    False,
+                    0.,
+                    "v0-Q",
+                    "png"
+                ],
+            ],
+        cache_examples = False,
+    )
+
     with gr.Row():
         gr.Markdown(claim_md)
-        event_id = gr.Textbox(label="Event ID", value="", visible=False)
     
     denoise_button.click(fn = check, inputs = [
         input_image