Adjusted hyperparameters
Browse files
app.py
CHANGED
|
@@ -100,7 +100,7 @@ with gr.Blocks() as demo:
|
|
| 100 |
gr.Markdown("# Noisy Image Restoration with Diffusion Models")
|
| 101 |
|
| 102 |
with gr.Row():
|
| 103 |
-
T = gr.Slider(4, 200, value=
|
| 104 |
stopping_sigma = gr.Slider(0.1, 5.0, value=0.1, step=0.1, label="Stopping Sigma (c)")
|
| 105 |
noise_sigma = gr.Slider(0, 0.6, value=0.05, step=0.01, label="Measurement Noise Sigma (σ_y)")
|
| 106 |
|
|
@@ -112,7 +112,7 @@ with gr.Blocks() as demo:
|
|
| 112 |
|
| 113 |
operator_select = gr.Dropdown(
|
| 114 |
choices=["Box Inpainting", "Random Inpainting", "Super Resolution", "Gaussian Deblur"],
|
| 115 |
-
value="
|
| 116 |
label="Select Task"
|
| 117 |
)
|
| 118 |
|
|
|
|
| 100 |
gr.Markdown("# Noisy Image Restoration with Diffusion Models")
|
| 101 |
|
| 102 |
with gr.Row():
|
| 103 |
+
T = gr.Slider(4, 200, value=25, step=1, label="Number of Inference Steps (T)")
|
| 104 |
stopping_sigma = gr.Slider(0.1, 5.0, value=0.1, step=0.1, label="Stopping Sigma (c)")
|
| 105 |
noise_sigma = gr.Slider(0, 0.6, value=0.05, step=0.01, label="Measurement Noise Sigma (σ_y)")
|
| 106 |
|
|
|
|
| 112 |
|
| 113 |
operator_select = gr.Dropdown(
|
| 114 |
choices=["Box Inpainting", "Random Inpainting", "Super Resolution", "Gaussian Deblur"],
|
| 115 |
+
value="Random Inpainting",
|
| 116 |
label="Select Task"
|
| 117 |
)
|
| 118 |
|