Add log

app.py

@@ -164,14 +164,15 @@ def stage2_process(
     print("spt_linear_CFG: " + str(spt_linear_CFG))
     print("spt_linear_s_stage2: " + str(spt_linear_s_stage2))
     print("model_select: " + str(model_select))
-    print("output_format: " + str(output_format))
     print("GPU time allocation: " + str(allocation) + " min")
+    print("output_format: " + str(output_format))
 
     if output_format == "input":
         if noisy_image is None:
             output_format = "png"
         else:
             output_format = re.sub(r"^.*\.([^\.]+)$", "\1", noisy_image)
+    print("final output_format: " + str(output_format))
 
     if prompt is None:
         prompt = ""
@@ -612,7 +613,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "input",
+                    "png",
                     5
                 ],
                 [
@@ -643,7 +644,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "input",
+                    "png",
                     4
                 ],
                 [
@@ -674,7 +675,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "input",
+                    "png",
                     4
                 ],
                 [
@@ -705,7 +706,7 @@ with gr.Blocks() as interface:
                     False,
                     0.,
                     "v0-Q",
-                    "input",
+                    "png",
                     4
                 ],
             ],