updated app

app.py

@@ -318,8 +318,8 @@ if uploaded_file is not None:
         rankedAttr = rankedAttr.detach().cpu().numpy()[0][0]
         rankedAttr = gray2rgb(rankedAttr)
         mplotfig, _ = visualize_image_attr(rankedAttr, origImgNP, method='blended_heat_map', cmap='RdYlGn')
-        fig = mplotfig.figure(figsize=(8,8))
-        st.pyplot(fig)
+        # fig = mplotfig.figure(figsize=(8,8))
+        st.pyplot(mplotfig)
         # mplotfig.savefig(outputDir + '{}_shapley_gl.png'.format(nameNoExt))
         # mplotfig.clear()
         # plt.close(mplotfig)