Meloo commited on
Commit
ada00c7
1 Parent(s): 5f0817d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -12,6 +12,7 @@ from PIL import Image
12
  from utils.download_url import load_file_from_url
13
  from utils.color_fix import wavelet_reconstruction
14
  from models.safmn_arch import SAFMN
 
15
 
16
 
17
  ########################################## Gradio inference ###################################
@@ -214,10 +215,10 @@ css = """
214
  demo = gr.Interface(
215
  fn=inference,
216
  inputs=[
217
- gr.Image(type="pil", label="Input", value="real_testdata/004.png"),
218
- gr.Number(default=2, label="Upscaling factor (up to 4)"),
219
- gr.Checkbox(default=False, label="Memory-efficient inference"),
220
- gr.Checkbox(default=False, label="Color correction"),
221
  ],
222
  outputs=ImageSlider(label="Super-Resolved Image",
223
  type="pil",
 
12
  from utils.download_url import load_file_from_url
13
  from utils.color_fix import wavelet_reconstruction
14
  from models.safmn_arch import SAFMN
15
+ from gradio_imageslider import ImageSlider
16
 
17
 
18
  ########################################## Gradio inference ###################################
 
215
  demo = gr.Interface(
216
  fn=inference,
217
  inputs=[
218
+ gr.Image(value="real_testdata/004.png", type="pil", label="Input"),
219
+ gr.Number(value=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
  ],
223
  outputs=ImageSlider(label="Super-Resolved Image",
224
  type="pil",