10 min allocation

app.py

@@ -137,9 +137,9 @@ def stage2_process(
     output_format,
     allocation
 ):
-    print("noisy_image: " + str(noisy_image))
+    print("noisy_image:\n" + str(noisy_image))
+    print("denoise_image:\n" + str(denoise_image))
     print("rotation: " + str(rotation))
-    print("denoise_image: " + str(denoise_image))
     print("prompt: " + str(prompt))
     print("a_prompt: " + str(a_prompt))
     print("n_prompt: " + str(n_prompt))
@@ -315,7 +315,7 @@ def restore(
     allocation
 ):
     start = time.time()
-    print('stage2_process ==>>')
+    print('restore ==>>')
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
@@ -360,7 +360,7 @@ def restore(
     # All the results have the same size
     result_height, result_width, result_channel = np.array(results[0]).shape
 
-    print('<<== stage2_process')
+    print('<<== restore')
     end = time.time()
     secondes = int(end - start)
     minutes = math.floor(secondes / 60)
@@ -474,8 +474,8 @@ with gr.Blocks() as interface:
         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)
         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)
-        allocation = gr.Radio([["1 min", 1], ["2 min", 2], ["3 min", 3], ["4 min", 4], ["5 min", 5], ["6 min", 6], ["7 min", 7], ["8 min", 8]], label="GPU allocation time", info="lower=May abort run, higher=Quota penalty for next runs", value=6, interactive=True)
-        output_format = gr.Radio([["As input", "input"], ["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"]], label="Image format for result", info="File extention", value="png", interactive=True)
+        allocation = gr.Radio([["1 min", 1], ["2 min", 2], ["3 min", 3], ["4 min", 4], ["5 min", 5], ["6 min", 6], ["7 min", 7], ["8 min", 8], ["9 min (discouraged)", 9], ["10 min (discouraged)", 10]], label="GPU allocation time", info="lower=May abort run, higher=Quota penalty for next runs", value=6, interactive=True)
+        output_format = gr.Radio([["As input", "input"], ["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"]], label="Image format for result", info="File extention", value="input", interactive=True)
 
     with gr.Accordion("Pre-denoising (optional)", open=False):
         gamma_correction = gr.Slider(label="Gamma Correction", info = "lower=lighter, higher=darker", minimum=0.1, maximum=2.0, value=1.0, step=0.1)
@@ -612,7 +612,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "png",
+                    "input",
                     5
                 ],
                 [
@@ -643,7 +643,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "png",
+                    "input",
                     4
                 ],
                 [
@@ -674,7 +674,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "webp",
+                    "input",
                     4
                 ],
                 [
@@ -705,7 +705,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "webp",
+                    "input",
                     4
                 ],
             ],