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"),
|