Spaces:
Duplicated from doevent/Face-Real-ESRGAN

bijoymirza99
/

Face-Real-ESRGAN

Runtime error

App Files Files Community

Face-Real-ESRGAN / app.py

bijoymirza99's picture

Create app.py

b0ffb5f verified 3 months ago

raw history blame contribute delete

No virus

419 Bytes

	import torch
	from PIL import Image
	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)

	sr_image.save('results/sr_image.png')