Log statistics

app.py

@@ -413,16 +413,13 @@ def restore(
     " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels."
     print(information)
     try:
-        print("Estimated minutes: " + str(math.log(result_width * result_height * input_width * input_height * edm_steps * num_samples)))
+        print("Initial resolution: " + str(input_width * input_height))
+        print("Final resolution: " + str(result_width * result_height))
+        print("edm_steps: " + str(edm_steps))
+        print("num_samples: " + str(num_samples))
+        print("Estimated minutes: " + str(math.log(result_width * result_height * input_width * input_height * edm_steps * num_samples) / 10))
     except Exception as e:
         print('Exception of Estimation')
-    try:
-        unique_name = str(uuid.uuid4()) + "." + output_format
-        image_copy = Image.fromarray(np.array(results[0]))
-        image_copy.save(unique_name)
-        print(unique_name)
-    except Exception as e:
-        print('Exception printing the path: ' + str(e))
 
     # Only one image can be shown in the slider
     return [noisy_image] + [results[0]], gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = results), gr.update(value = information, visible = True), gr.update(visible=True)
@@ -596,45 +593,6 @@ with gr.Blocks() as interface:
         result_gallery = gr.Gallery(label = 'Downloadable results', show_label = True, elem_id = "gallery1")
 
     gr.Examples(
-        run_on_click = True,
-        fn = stage2_process,
-	    inputs = [
-            input_image,
-            rotation,
-            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,
-            allocation
-        ],
-	    outputs = [
-            result_slider,
-            result_gallery,
-            restore_information,
-            reset_btn
-        ],
         examples = [
                 [
                     "./Examples/Example1.png",
@@ -761,6 +719,45 @@ with gr.Blocks() as interface:
                     4
                 ],
             ],
+        run_on_click = True,
+        fn = stage2_process,
+	    inputs = [
+            input_image,
+            rotation,
+            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,
+            allocation
+        ],
+	    outputs = [
+            result_slider,
+            result_gallery,
+            restore_information,
+            reset_btn
+        ],
         cache_examples = False,
     )