Eugene Siow commited on
Commit
d7b9b40
1 Parent(s): 9eb308e

Remove larger models to mitigate CPU memory limit errors.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -65,7 +65,8 @@ torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/result
65
  torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/bird_mini_d4_gaussian.bmp',
66
  'bird.bmp')
67
 
68
- models = ['EDSR-base', 'DRLN', 'EDSR', 'MDSR', 'A2N', 'PAN', 'AWSRN-BAM', 'MSRN']
 
69
  scales = [2, 3, 4]
70
  for model_name in models:
71
  for scale in scales:
@@ -86,7 +87,7 @@ gr.Interface(
86
  examples=[
87
  ['baby.bmp', 'x2', 'EDSR-base'],
88
  ['woman.bmp', 'x3', 'DRLN'],
89
- ['bird.bmp', 'x4', 'EDSR']
90
  ],
91
  enable_queue=True,
92
  allow_flagging=False,
 
65
  torch.hub.download_url_to_file('http://people.rennes.inria.fr/Aline.Roumy/results/images_SR_BMVC12/input_groundtruth/bird_mini_d4_gaussian.bmp',
66
  'bird.bmp')
67
 
68
+ # models = ['EDSR-base', 'DRLN', 'EDSR', 'MDSR', 'A2N', 'PAN', 'AWSRN-BAM', 'MSRN']
69
+ models = ['EDSR-base', 'DRLN', 'A2N', 'PAN', 'AWSRN-BAM', 'MSRN']
70
  scales = [2, 3, 4]
71
  for model_name in models:
72
  for scale in scales:
 
87
  examples=[
88
  ['baby.bmp', 'x2', 'EDSR-base'],
89
  ['woman.bmp', 'x3', 'DRLN'],
90
+ ['bird.bmp', 'x4', 'MSRN']
91
  ],
92
  enable_queue=True,
93
  allow_flagging=False,