Adjusted hyperparameters
Browse files
app.py
CHANGED
|
@@ -100,9 +100,9 @@ with gr.Blocks() as demo:
|
|
| 100 |
gr.Markdown("# Noisy Image Restoration with Diffusion Models")
|
| 101 |
|
| 102 |
with gr.Row():
|
| 103 |
-
T = gr.Slider(
|
| 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="Noise Sigma")
|
| 106 |
|
| 107 |
image_select = gr.Dropdown(
|
| 108 |
choices=["CelebA HQ 1", "CelebA HQ 2", "CelebA HQ 3", "LSUN Church"],
|
|
|
|
| 100 |
gr.Markdown("# Noisy Image Restoration with Diffusion Models")
|
| 101 |
|
| 102 |
with gr.Row():
|
| 103 |
+
T = gr.Slider(4, 200, value=50, 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 |
|
| 107 |
image_select = gr.Dropdown(
|
| 108 |
choices=["CelebA HQ 1", "CelebA HQ 2", "CelebA HQ 3", "LSUN Church"],
|