Eugene Siow commited on
Commit
89d3a8e
·
1 Parent(s): fd32bd4

Fix runtime error by temporarily removing some models.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -51,14 +51,15 @@ torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/result
51
  'baby.bmp')
52
  torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/woman_mini_d3_gaussian.bmp',
53
  'woman.bmp')
 
 
54
 
55
  gr.Interface(
56
  inference,
57
  [
58
  gr.inputs.Image(type="pil", label="Input"),
59
  gr.inputs.Radio(["x2", "x3", "x4"], label='scale'),
60
- gr.inputs.Dropdown(choices=['EDSR-base', 'EDSR', 'MSRN', 'MDSR', 'AWSRN-BAM', 'A2N', 'CARN', 'PAN', 'HAN',
61
- 'DRLN', 'RCAN'],
62
  label='Model')
63
  ],
64
  gr.outputs.Image(type="file", label="Output"),
 
51
  'baby.bmp')
52
  torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/woman_mini_d3_gaussian.bmp',
53
  'woman.bmp')
54
+ # models = ['EDSR-base', 'EDSR', 'MSRN', 'MDSR', 'AWSRN-BAM', 'A2N', 'CARN', 'PAN', 'HAN', 'DRLN', 'RCAN']
55
+ models = ['EDSR-base', 'MSRN', 'MDSR', 'AWSRN-BAM', 'A2N', 'CARN', 'PAN']
56
 
57
  gr.Interface(
58
  inference,
59
  [
60
  gr.inputs.Image(type="pil", label="Input"),
61
  gr.inputs.Radio(["x2", "x3", "x4"], label='scale'),
62
+ gr.inputs.Dropdown(choices=models,
 
63
  label='Model')
64
  ],
65
  gr.outputs.Image(type="file", label="Output"),