Update app.py

convert image to PIL format

app.py

@@ -24,7 +24,11 @@ with open('labels.txt', "r") as f:
     labels = [line.strip() for line in f.readlines()]
 
 
-def classify(image):
+def classify(image_raw):
+
+    ## convert image to PIL format
+    image = Image.new(image_raw)
+    
     ## preprocessing
     # we need a transform step to normalise the pictures
     transform = T.Compose([T.Resize(256), T.CenterCrop(224), T.ToTensor(),