Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,6 +26,7 @@ from realesrgan.archs.srvgg_arch import SRVGGNetCompact
|
|
26 |
# global variables
|
27 |
last_file = None
|
28 |
img_mode = "RGB"
|
|
|
29 |
|
30 |
# Upscale function
|
31 |
def upscale(img, choice):
|
@@ -54,6 +55,10 @@ def upscale(img, choice):
|
|
54 |
model_path = os.path.join('weights', '2xNomosUni_compact_otf_medium.pth')
|
55 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16, upscale=2, act_type='prelu')
|
56 |
|
|
|
|
|
|
|
|
|
57 |
# Restorer Class
|
58 |
upsampler = RealESRGANer(
|
59 |
scale=2,
|
|
|
26 |
# global variables
|
27 |
last_file = None
|
28 |
img_mode = "RGB"
|
29 |
+
choice= "2x Fast Upscale"
|
30 |
|
31 |
# Upscale function
|
32 |
def upscale(img, choice):
|
|
|
55 |
model_path = os.path.join('weights', '2xNomosUni_compact_otf_medium.pth')
|
56 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16, upscale=2, act_type='prelu')
|
57 |
|
58 |
+
else:
|
59 |
+
model_path = os.path.join('weights', '2xNomosUni_compact_multijpg_ldl.pth')
|
60 |
+
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16, upscale=2, act_type='prelu')
|
61 |
+
|
62 |
# Restorer Class
|
63 |
upsampler = RealESRGANer(
|
64 |
scale=2,
|