Update handler.py

handler.py

@@ -14,14 +14,14 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 class EndpointHandler():
 
-    def __init__(self, path="", esrgan_model_path=""):
+    def __init__(self, path=""):
         # Load the StableDiffusionPipeline
         self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float32)
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.pipe = self.pipe.to(device)
 
         # Load the ESRGAN state dictionary
-        checkpoint = torch.load(esrgan_model_path)
+        checkpoint = torch.load("RealESRGAN_x4plus_anime_6B.pth")
 
         # Check if 'params_ema' is in the keys and filter the state_dict
         if "params_ema" in checkpoint: