Update app.py

Added transform to the code

app.py

@@ -15,6 +15,17 @@ import torchvision.transforms.functional as TF
 from PIL import Image
 import numpy as np
 
+def image_transform(image):
+    if isinstance(image, str):
+        # If image is a path to a file, open it using PIL
+        image = Image.open(image).convert('RGB')
+    else:
+        # If image is a NumPy array, convert it to a PIL image
+        image = Image.fromarray(image.astype('uint8'), 'RGB')
+    # Apply the same transformations as before
+    image = transform(image).unsqueeze(0)
+    return image.to(device)
+
 
 #Defining the predict function
 def style_transfer(cont_img,styl_img):