Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,12 +13,12 @@ sr_pipe4x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=4)
|
|
13 |
def inference(image, size):
|
14 |
if image is None:
|
15 |
raise gr.Error("Image not uploaded")
|
16 |
-
r_image = Image.open(image)
|
17 |
|
18 |
if size == '2x':
|
19 |
-
result = sr_pipe2x(
|
20 |
else:
|
21 |
-
result = sr_pipe4x(
|
22 |
|
23 |
print(f"Image size ({device}): {size} ... OK")
|
24 |
return result
|
@@ -30,7 +30,7 @@ article = "<div style='text-align: center;'>Twitter <a href='https://twitter.com
|
|
30 |
|
31 |
|
32 |
gr.Interface(inference,
|
33 |
-
[gr.Image(type="
|
34 |
gr.Radio(['2x', '4x'],
|
35 |
type="value",
|
36 |
value='2x',
|
|
|
13 |
def inference(image, size):
|
14 |
if image is None:
|
15 |
raise gr.Error("Image not uploaded")
|
16 |
+
# r_image = Image.open(image)
|
17 |
|
18 |
if size == '2x':
|
19 |
+
result = sr_pipe2x(image)
|
20 |
else:
|
21 |
+
result = sr_pipe4x(image)
|
22 |
|
23 |
print(f"Image size ({device}): {size} ... OK")
|
24 |
return result
|
|
|
30 |
|
31 |
|
32 |
gr.Interface(inference,
|
33 |
+
[gr.Image(type="pil"),
|
34 |
gr.Radio(['2x', '4x'],
|
35 |
type="value",
|
36 |
value='2x',
|