Update inference_realesrgan.py

inference_realesrgan.py

@@ -43,7 +43,7 @@ def main():
 
     # determine models according to model names
     args.model_name = args.model_name.split('.')[0]
-    if args.model_name in ['RealESRGAN_x4plus', 'RealESRNet_x4plus']:  # x4 RRDBNet model
+    if args.model_name in ['RealESRGAN_x4plus', 'RealESRNet_x4plus', 'RealESRGAN_x4plus_super.pth']:  # x4 RRDBNet model
         model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
         netscale = 4
     elif args.model_name in ['RealESRGAN_x4plus_anime_6B']:  # x4 RRDBNet model with 6 blocks