Update app.py

app.py

@@ -126,15 +126,13 @@ class LazyRealESRGAN:
         if self.model is None:
             self.model = RealESRGAN(self.device, scale=self.scale)
             self.model.load_weights(f'models/upscalers/RealESRGAN_x{self.scale}.pth', download=False)
-
+            self.model  = torch.compile(self.model)
     def predict(self, img):
         self.load_model()
         return self.model.predict(img)
 
 lazy_realesrgan_x2 = LazyRealESRGAN(device, scale=2)
-lazy_realesrgan_x2  = torch.compile(lazy_realesrgan_x2)
 lazy_realesrgan_x4 = LazyRealESRGAN(device, scale=4)
-lazy_realesrgan_x4  = torch.compile(lazy_realesrgan_x4)
 
 @timer_func
 def resize_and_upscale(input_image, resolution):