Update app.py

app.py

@@ -15,7 +15,7 @@ device = "cpu"
 model = Generator().eval().to(device)
 model.load_state_dict(torch.load("face_paint_512_v2_0.pt"))
 
-from PIL import Image
+from PIL import Image, ImageOps
 from torchvision.transforms.functional import to_tensor, to_pil_image
 import gradio as gr
 
@@ -55,7 +55,7 @@ import scipy.ndimage
 import requests
 
 def inference(image):
-  img = image
+  img = ImageOps.contain(image, (512,512))
   out = face2paint(img, 512)
   return out
   
@@ -65,4 +65,4 @@ description = "Gradio demo for AnimeGanv2 Face Portrait v2. To use it, simply up
 article = "<p style='text-align: center'><a href='https://github.com/bryandlee/animegan2-pytorch' target='_blank'>Github Repo</a></p>"
 
 examples=[['groot.jpeg']]
-gr.Interface(inference, gr.inputs.Image(type="pil",shape=(512,512)), gr.outputs.Image(type="pil"),title=title,description=description,article=article,examples=examples,enable_queue=True).launch()
+gr.Interface(inference, gr.inputs.Image(type="pil"), gr.outputs.Image(type="pil"),title=title,description=description,article=article,examples=examples,enable_queue=True).launch()