Update app.py

app.py

@@ -52,6 +52,7 @@ def get_gradcam_image(input_tensor, label, target_layer):
 def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2, 3], transparency=0.5):
     input_image = transform(image=input_image)
     #aug(image=image)
+    input_image = input_image.unsqueeze(0)
     output = model(input_image)
     output = F.softmax(output.flatten(), dim=-1)