Update app.py

app.py

@@ -87,7 +87,7 @@ input_interface = gr.Interface(inference,
                         gr.Slider(-2, -1, value=-2, step=1, label="Which Layer?"),
                         gr.Slider(1, 10, value=3, step=1, label="How many top confidence classes to be shown?")],
                 outputs=[gr.Label(),
-                         gr.Image(shape=(32, 32), label="Model Prediction").style(width=128, height=128)],
+                         gr.Image(shape=(32, 32), label="Model Prediction").style(width=512, height=512)],
                 description=description1,examples=examples)
 
 description2 = "Displays misclassified image of the model"