Update app.py

app.py

@@ -161,7 +161,7 @@ demo = gr.Interface(
         ),
     ],
     outputs=[
-        gr.Image(shape=(32, 32)).style(width=256, height=256),
+        gr.Image(shape=(32, 32), label="CAM Output for the Chosen Class").style(width=256, height=256),
         gr.Label(label="Top Classes"),
     ],
     examples=[[os.path.join("./samples/", f)] for f in os.listdir("./samples/")]