chore: checking with plt obj

app.py

@@ -47,5 +47,4 @@ with st.spinner("Generating the heat maps... HOLD ON!"):
 
 # Show the attention maps
 st.title("Attention 🔥 Maps")
-image = Image.frombytes('RGB', plt.canvas.get_width_height(), plt.canvas.tostring_rgb())
-st.image(image, caption="Attention Heat Maps")
+st.image(plt, caption="Attention Heat Maps")