add steps

app_upscale.py

@@ -26,10 +26,11 @@ DEFAULT_CATEGORY = "face"
 
 def create_demo() -> gr.Blocks:
 
-    @spaces.GPU(duration=15)
+    @spaces.GPU(duration=30)
     def upscale_image(
         input_image: Image,
         prompt: str,
+        num_inference_steps: int = 10,
     ):
         time_cost_str = ''
         run_task_time = 0
@@ -56,6 +57,7 @@ def create_demo() -> gr.Blocks:
             with gr.Column():
                 input_image_prompt = gr.Textbox(lines=1, label="Input Image Prompt", value=DEFAULT_SRC_PROMPT)
             with gr.Column():
+                num_inference_steps = gr.Number(label="Num Inference Steps", value=5)
                 g_btn = gr.Button("Upscale Image")
                 
         with gr.Row():
@@ -78,7 +80,7 @@ def create_demo() -> gr.Blocks:
             outputs=[origin_area_image, croper],
         ).success(
             fn=upscale_image,
-            inputs=[origin_area_image, input_image_prompt],
+            inputs=[origin_area_image, input_image_prompt, num_inference_steps],
             outputs=[upscaled_image, generated_cost],
         ).success(
             fn=restore_result,