Spaces:

Guldeniz
/

aerial-to-map

Runtime error

aerial-to-map / app.py

Update app.py

d281e78 about 2 years ago

No virus

668 Bytes

	import gradio as gr
	from torchvision.transforms import Compose, Resize, ToTensor, Normalize
	from PIL import Image
	from torchvision.utils import save_image
	from huggan.pytorch.pix2pix.modeling_pix2pix import GeneratorUNet
	6
	transform = Compose(
	[
	Resize((256, 256), Image.BICUBIC),
	ToTensor(),
	Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
	]
	)
	model = GeneratorUNet.from_pretrained('Guldeniz/pix2pix_maps')

	def predict_fn(img):
	inp = transform(img).unsqueeze(0)
	out = model(inp)
	save_image(out, 'out.png', normalize=True)
	return 'out.png'

	gr.Interface(predict_fn, inputs=gr.inputs.Image(type='pil'), outputs='image').launch()