ali-ghamdan commited on
Commit
763bb4e
1 Parent(s): 96b0185

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -4,7 +4,7 @@ import colorizers as c
4
 
5
  from colorizers.util import postprocess_tens, preprocess_img
6
 
7
- def interface(image, model: str):
8
  if model == "eccv16":
9
  img = c.eccv16(pretrained=True).eval()
10
  else:
@@ -27,11 +27,10 @@ gr.Interface(
27
  gr.components.Radio(
28
  ["eccv16", "siggraph17"],
29
  type="value",
30
- default_selected="eccv16",
31
  label="model"
32
  )
33
  ],
34
  [
35
- gr.outputs.Image(label="Output")
36
  ]
37
- )
 
4
 
5
  from colorizers.util import postprocess_tens, preprocess_img
6
 
7
+ def interface(image, model: str = "eccv16"):
8
  if model == "eccv16":
9
  img = c.eccv16(pretrained=True).eval()
10
  else:
 
27
  gr.components.Radio(
28
  ["eccv16", "siggraph17"],
29
  type="value",
 
30
  label="model"
31
  )
32
  ],
33
  [
34
+ gr.components.Image(label="output")
35
  ]
36
+ ).launch()