Spaces:
Runtime error
Runtime error
Eugene Siow
commited on
Commit
·
89d3a8e
1
Parent(s):
fd32bd4
Fix runtime error by temporarily removing some models.
Browse files
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=
|
| 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"),
|