File format

gradio_demo.py

@@ -141,7 +141,8 @@ def stage2_process(
     linear_s_stage2,
     spt_linear_CFG,
     spt_linear_s_stage2,
-    model_select
+    model_select,
+    output_format
 ):
     start = time.time()
     print('Start stage2_process')
@@ -210,9 +211,9 @@ def stage2_process(
     print('End stage2_process')
     end = time.time()
     secondes = int(end - start)
-    minutes = secondes // 60
+    minutes = math.floor(secondes / 60)
     secondes = secondes - (minutes * 60)
-    hours = minutes // 60
+    hours = math.floor(minutes / 60)
     minutes = minutes - (hours * 60)
     information = ("Start the process again if you want a different result. " if randomize_seed else "") + \
     "The new image resolution is " + str(result_width) + \
@@ -224,7 +225,7 @@ def stage2_process(
     str(secondes) + " sec."
 
     # Only one image can be shown in the slider
-    return [noisy_image] + [results[0]], [noisy_image] + results, gr.update(value = information, visible = True), event_id
+    return [noisy_image] + [results[0]], gr.update(format = output_format, value = [noisy_image] + results), gr.update(value = information, visible = True), event_id
 
 def load_and_reset(param_setting):
     print('Start load_and_reset')
@@ -329,6 +330,7 @@ with gr.Blocks(title="SUPIR") as interface:
                           lines=3)
     with gr.Group():
         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.")
+    output_format = gr.Radio(["png", "webp", "jpeg", "gif", "bmp"], label="Image format for result", info="File extention", value="png", interactive=True)
 
     with gr.Accordion("Pre-denoising (optional)", open=False):
         gamma_correction = gr.Slider(label="Gamma Correction", minimum=0.1, maximum=2.0, value=1.0, step=0.1)
@@ -458,7 +460,8 @@ with gr.Blocks(title="SUPIR") as interface:
         linear_s_stage2,
         spt_linear_CFG,
         spt_linear_s_stage2,
-        model_select
+        model_select,
+        output_format
     ], outputs = [
         result_slider,
         result_gallery,