Meloo commited on
Commit
b82bdb5
1 Parent(s): 9bf0a2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -15,7 +15,6 @@ from models.safmn_arch import SAFMN
15
  from gradio_imageslider import ImageSlider
16
 
17
 
18
- ########################################## Gradio inference ###################################
19
  pretrain_model_url = {
20
  'safmn_x2': 'https://github.com/sunny2109/SAFMN/releases/download/v0.1.0/SAFMN_L_Real_LSDIR_x2-v2.pth',
21
  'safmn_x4': 'https://github.com/sunny2109/SAFMN/releases/download/v0.1.0/SAFMN_L_Real_LSDIR_x4-v2.pth',
@@ -217,7 +216,7 @@ demo = gr.Interface(
217
  fn=inference,
218
  inputs=[
219
  gr.Image(value="real_testdata/004.png", type="pil", label="Input"),
220
- gr.Number(default=10, label="Upscaling factor (up to 4)"),
221
  gr.Checkbox(value=False, label="Memory-efficient inference"),
222
  gr.Checkbox(value=False, label="Color correction"),
223
  ],
 
15
  from gradio_imageslider import ImageSlider
16
 
17
 
 
18
  pretrain_model_url = {
19
  'safmn_x2': 'https://github.com/sunny2109/SAFMN/releases/download/v0.1.0/SAFMN_L_Real_LSDIR_x2-v2.pth',
20
  'safmn_x4': 'https://github.com/sunny2109/SAFMN/releases/download/v0.1.0/SAFMN_L_Real_LSDIR_x4-v2.pth',
 
216
  fn=inference,
217
  inputs=[
218
  gr.Image(value="real_testdata/004.png", type="pil", label="Input"),
219
+ gr.Number(default=2, label="Upscaling factor (up to 4)"),
220
  gr.Checkbox(value=False, label="Memory-efficient inference"),
221
  gr.Checkbox(value=False, label="Color correction"),
222
  ],