myshop-capsule
/

image-super-resolution

Model card Files Files and versions

justin-shopcapsule commited on Aug 24, 2023

Commit

39bb242

·

1 Parent(s): 83ec49b

Update handler.py

use the 2x scale model

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -19,8 +19,8 @@ from RealESRGAN import RealESRGAN
 class EndpointHandler():
     def __init__(self, path="."):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.model = RealESRGAN(self.device, scale=4)
-        self.model.load_weights('/repository/RealESRGAN_x4.pth', download=True)
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 class EndpointHandler():
     def __init__(self, path="."):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.model = RealESRGAN(self.device, scale=2)
+        self.model.load_weights('/repository/RealESRGAN_x2.pth', download=True)
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: