Fix imports

RealESRGAN/__init__.py

+from .model import RealESRGAN
\ No newline at end of file

RealESRGAN/model.py

@@ -4,8 +4,8 @@ from PIL import Image
 import numpy as np
 import cv2
 
-from rrdbnet_arch import RRDBNet
-from utils import *
+from .rrdbnet_arch import RRDBNet
+from .utils import *
 
 
 class RealESRGAN:

RealESRGAN/rrdbnet_arch.py

@@ -2,7 +2,7 @@ import torch
 from torch import nn as nn
 from torch.nn import functional as F
 
-from arch_utils import default_init_weights, make_layer, pixel_unshuffle
+from .arch_utils import default_init_weights, make_layer, pixel_unshuffle
 
 
 class ResidualDenseBlock(nn.Module):

main.py

@@ -2,7 +2,7 @@ import os
 import torch
 from PIL import Image
 import numpy as np
-from realesrgan import RealESRGAN
+from RealESRGAN import RealESRGAN
 
 
 def main() -> int:
@@ -13,7 +13,6 @@ def main() -> int:
         image = Image.open(f"inputs/{image}").convert('RGB')
         sr_image = model.predict(image)
         sr_image.save(f'results/{i}.png')
-    return 1 
 
 
 if __name__ == '__main__':