Rainy-hh
/

Real-ESRGAN

Inference Endpoints

Model card Files Files and versions Community

Rainy commited on Aug 18, 2023

Commit

57b1ce0

•

1 Parent(s): c414443

:tada: first commit

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ import numpy as np
 from RealESRGAN import RealESRGAN
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = RealESRGAN(device, scale=4)
-model.load_weights('weights/RealESRGAN_x4.pth', download=True)
 path_to_image = 'inputs/lr_image.png'
 image = Image.open(path_to_image).convert('RGB')
 sr_image = model.predict(image)

 from RealESRGAN import RealESRGAN
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = RealESRGAN(device, scale=4)
+model.load_weights('weights/RealESRGAN_x4plus.pth', download=True)
 path_to_image = 'inputs/lr_image.png'
 image = Image.open(path_to_image).convert('RGB')
 sr_image = model.predict(image)