updated app.py

app.py

@@ -61,7 +61,7 @@ def show_misclassified_images_wrapper(num_images=10, use_gradcam=False, gradcam_
 
     return model.show_misclassified_images(num_images, use_gradcam, gradcam_layer, transparency)
 
-description1 = "Input any image to test the models prediction. Currently the model only supports the following classes - plane, car, bird, cat, deer, dog, frog, horse, ship, truck."
+description1 = "Test the model's prediction. Currently the model only supports the following classes - plane, car, bird, cat, deer, dog, frog, horse, ship, truck."
 
 
 # Define the full path to the images folder
@@ -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=512, height=512)],
+                         gr.Image(shape=(32, 32), label="Model Prediction").style(width=300, height=300)],
                 description=description1,examples=examples)
 
 description2 = "Displays misclassified image of the model"