apply num_inference_steps

app_upscale.py

@@ -35,7 +35,11 @@ def create_demo() -> gr.Blocks:
         time_cost_str = ''
         run_task_time = 0
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-        upscaled_image = upscale_pipe(prompt=prompt, image=input_image).images[0]
+        upscaled_image = upscale_pipe(
+            prompt=prompt, 
+            image=input_image,
+            num_inference_steps=num_inference_steps,
+        ).images[0]
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
         
         return upscaled_image, time_cost_str