Square root

app.py

@@ -201,11 +201,17 @@ def restore_in_Xmin(
     print("GPU time allocation: " + str(allocation) + " min")
     print("output_format: " + str(output_format))
 
+    input_format = re.sub(r"^.*\.([^\.]+)$", r"\1", noisy_image)
+
+    if input_format not in ['png', 'webp', 'jpg', 'jpeg', 'bmp']:
+        gr.Warning('Invalid image format. Please first convert into *.png, *.webp, *.jpg, *.jpeg or *.bmp.')
+        return None, None, None, None
+
     if output_format == "input":
         if noisy_image is None:
             output_format = "png"
         else:
-            output_format = re.sub(r"^.*\.([^\.]+)$", r"\1", noisy_image)
+            output_format = input_format
     print("final output_format: " + str(output_format))
 
     if prompt is None:
@@ -417,7 +423,8 @@ def restore(
         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))
+        print("downscale: " + str(downscale))
+        print("Estimated minutes: " + str((result_width * result_height * input_width * input_height * edm_steps * num_samples)**(1/2)))
     except Exception as e:
         print('Exception of Estimation')
 
@@ -515,7 +522,7 @@ with gr.Blocks() as interface:
     """)
     gr.HTML(title_html)
 
-    input_image = gr.Image(label="Input (*.png, *.webp, *.jpeg, *.gif, *.bmp)", show_label=True, type="filepath", height=600, elem_id="image-input")
+    input_image = gr.Image(label="Input (*.png, *.webp, *.jpeg, *.jpg, *.bmp)", show_label=True, type="filepath", height=600, elem_id="image-input")
     rotation = gr.Radio([["No rotation", 0], ["⤵ Rotate +90°", 90], ["↩ Return 180°", 180], ["⤴ Rotate -90°", -90]], label="Orientation correction", info="Will apply the following rotation before restoring the image; the AI needs a good orientation to understand the content", value=0, interactive=True, visible=False)
     with gr.Group():
         prompt = gr.Textbox(label="Image description", info="Help the AI understand what the image represents; describe as much as possible, especially the details we can't see on the original image; you can write in any language", value="", placeholder="A 33 years old man, walking, in the street, Santiago, morning, Summer, photorealistic", lines=3)