fix generate_size

app_base.py

@@ -72,6 +72,7 @@ def create_demo() -> gr.Blocks:
             upscale_steps=upscale_steps,
         )
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
+        enhanced_image = enhanced_image.resize((generate_size, generate_size))
 
         return enhanced_image, res_image, time_cost_str