Update app.py

app.py

@@ -126,8 +126,8 @@ description_with_icon = f"""
 demo = gr.Interface(processed_image,
     inputs=[gr.Image(label="Input Image"),],
     outputs=[
-        gr.Plot(label="Output Plot",style="background-color: #f3f5f1;"), 
-        gr.Image(shape=(32, 32), label="Model Prediction"),
+        gr.Plot(label="Output with Classes",), 
+        gr.Image(shape=(32, 32), label="GradCAM Output"),
     ],
     title=title_with_icon,
     description=description_with_icon,