Phips commited on
Commit
9cd401a
1 Parent(s): edc7fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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,