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, ImageOps
+from PIL import Image
 from torchvision.transforms.functional import to_tensor, to_pil_image
 import gradio as gr
 
@@ -54,11 +54,14 @@ import scipy.ndimage
 
 import requests
 
-def inference(image):
-  img = ImageOps.contain(image, (512,512))
-  out = face2paint(img, 512)
-  return out
-  
+def inference(img):
+    basewidth = 512
+    wpercent = (basewidth/float(img.size[0]))
+    hsize = int((float(img.size[1])*float(wpercent)))
+    img = img.resize((basewidth,hsize), Image.ANTIALIAS)
+    out = face2paint(img, 512)
+    return out
+      
   
 title = "Animeganv2"
 description = "Gradio demo for AnimeGanv2 Face Portrait v2. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."