Update app.py

app.py

@@ -50,7 +50,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).unsqueeze(0)
+    input_image = transform(image=input_image)
     #aug(image=image)
     output = model(input_image)
     output = F.softmax(output.flatten(), dim=-1)