Update app.py

app.py

@@ -326,37 +326,53 @@ If you have any questions, please feel free to reach me out at <b>iceclearwjy@gm
 <center><img src='https://visitor-badge.laobi.icu/badge?page_id=IceClear/StableSR' alt='visitors'></center>
 """
 
-demo = gr.Interface(
-	inference, [
-		gr.inputs.Image(type="filepath", label="Input"),
-		gr.inputs.Number(default=1, label="Rescaling_Factor (Large images require huge time)"),
-		gr.Slider(0, 1, value=0.5, step=0.01, label='CFW_Fidelity (0 for better quality, 1 for better identity)'),
-		gr.inputs.Number(default=42, label="Seeds"),
-		gr.Dropdown(
-			choices=["512", "768v"],
-			value="512",
-			label="Model",
-			),
-		gr.Slider(10, 1000, value=200, step=1, label='Sampling timesteps for DDPM'),
-		gr.Dropdown(
-			choices=["none", "adain", "wavelet"],
-			value="adain",
-			label="Color_Correction",
-			),
-	], [
-		gr.outputs.Image(type="numpy", label="Output"),
-		gr.outputs.File(label="Download the output")
-	],
-	title=title,
-	description=description,
-	article=article,
-	examples=[
-		['./01.png', 4, 0.5, 42, "512", 200, "adain"],
-		['./02.png', 4, 0.5, 42, "512", 200, "adain"],
-		['./03.png', 4, 0.5, 42, "512", 200, "adain"],
-		['./04.png', 4, 0.5, 42, "512", 200, "adain"],
-		['./05.png', 4, 0.5, 42, "512", 200, "adain"]
-		]
+with gr.Blocks(title=title) as demo:
+	gr.Markdown(description)
+	gr.Markdown(article)
+
+	with gr.Row():
+		with gr.Column():
+			image = gr.Image(type="filepath", label="Input")
+			upscale = gr.Number(value=1, label="Rescaling_Factor (Large images require huge time)")
+			dec_w = gr.Slider(0, 1, value=0.5, step=0.01, label='CFW_Fidelity (0 for better quality, 1 for better identity)')
+			seed = gr.Number(value=42, label="Seeds")
+			model_type = gr.Dropdown(
+				choices=["512", "768v"],
+				value="512",
+				label="Model"
+			)
+			ddpm_steps = gr.Slider(10, 1000, value=200, step=1, label='Sampling timesteps for DDPM')
+			colorfix_type = gr.Dropdown(
+				choices=["none", "adain", "wavelet"],
+				value="adain",
+				label="Color_Correction"
+			)
+			run_btn = gr.Button("Run Inference")
+
+		with gr.Column():
+			output_image = gr.Image(type="numpy", label="Output")
+			output_file = gr.File(label="Download the output")
+
+	# Inference trigger
+	run_btn.click(
+		fn=inference,
+		inputs=[image, upscale, dec_w, seed, model_type, ddpm_steps, colorfix_type],
+		outputs=[output_image, output_file]
+	)
+
+	# Example section
+	gr.Examples(
+		examples=[
+			['./01.png', 4, 0.5, 42, "512", 200, "adain"],
+			['./02.png', 4, 0.5, 42, "512", 200, "adain"],
+			['./03.png', 4, 0.5, 42, "512", 200, "adain"],
+			['./04.png', 4, 0.5, 42, "512", 200, "adain"],
+			['./05.png', 4, 0.5, 42, "512", 200, "adain"]
+		],
+		fn=inference,
+		inputs=[image, upscale, dec_w, seed, model_type, ddpm_steps, colorfix_type],
+		outputs=[output_image, output_file],
+		cache_examples=True
 	)
 
 demo.queue(concurrency_count=1)