Update app.py

app.py

@@ -50,8 +50,8 @@ 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(input_image).unsqueeze(0)
-
+    input_image = transform(image=input_image).unsqueeze(0)
+    #aug(image=image)
     output = model(input_image)
     output = F.softmax(output.flatten(), dim=-1)
 
@@ -104,9 +104,9 @@ demo2 = gr.Interface(
     fn=show_incorrect,
     inputs=[
         gr.Number(value=10, minimum=1, maximum=50, label="No. of Misclassified Images", precision=0,
-                  info="How many misclassified examples do you want to view? (1 - 50)")
+                  info="How many misclassified examples do you want to view? (max 50)")
     ],
-    outputs=[gr.Gallery(label="Missclassified Images (Truth / Predicted)", columns=5)]
+    outputs=[gr.Gallery(label="Missclassified Images (Actual / Predicted)", columns=5)]
 )
 
 demo = gr.TabbedInterface([demo1, demo2], ["Classifier", "Misclassified Images"])