Update app.py

app.py

@@ -21,6 +21,7 @@ def inference(input_img, gradcam=False, num_gradcam=1, transparency=0.5, target_
               misclassified=False, num_misclassified=1, top_classes=3):
     input_img = np.array(input_img)
     org_img = input_img
+    input_img = Image.fromarray(input_img)
     transform = transforms.Compose([
         transforms.Resize((32, 32)),
         transforms.ToTensor(),