Update app.py

app.py

@@ -13,7 +13,7 @@ def inference(image):
     result = model.style_transfer(paths=[image.name])
     print(type(result[0]))
     print(result[0])
-    return Image.fromarray(np.uint8(result[0])).convert('RGB')[:,:,::-1]
+    return Image.fromarray(np.uint8(result[0])[:,:,::-1]).convert('RGB')
     
 title = "UGATIT-selfie2anime"
 description = "Gradio demo for DeOldify. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."