Update app.py

app.py

@@ -137,8 +137,8 @@ demo = gr.Interface(inference,
                               gr.Slider(0, 10, value = 0, step=1, label="How many top classes you want to see?")
                               ], 
                     outputs = [
-                              gr.Gallery(label="GradCAM Outputs", show_label=True, elem_id="gallery").style(columns=[2], rows=[2], object_fit="contain", height="auto"), 
-                              gr.Gallery(label="Misclassified Images", show_label=True, elem_id="gallery").style(columns=[2], rows=[2], object_fit="contain", height="auto"), 
+                              gr.Gallery(label="GradCAM Outputs", show_label=True, elem_id="gallery",columns=[2], rows=[2], object_fit="contain", height="auto"), 
+                              gr.Gallery(label="Misclassified Images", show_label=True, elem_id="gallery",columns=[2], rows=[2], object_fit="contain", height="auto"), 
                               gr.Label(num_top_classes=10, label = "Top classes")
                               ],
                     title = title,