vivjay30 commited on
Commit
f5a8182
·
1 Parent(s): 63d493a

Adjusted hyperparameters

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(10, 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="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"],