Update app.py

app.py

@@ -217,4 +217,20 @@ if uploaded_file is not None:
             table_data = metrics_df.values
             table_columns = metrics_df.columns.tolist()
             
-            table = ax.table(cellText=table_data, colLabels=table_columns, loc='center', cellLoc='center',
+            table = ax.table(cellText=table_data, colLabels=table_columns, loc='center', cellLoc='center', colLoc='center')
+            table.auto_set_font_size(False)
+            table.set_fontsize(10)
+            table.scale(1.2, 1.2)  # Adjust the scale for better appearance
+
+            # Save the table as a PNG file
+            png_file = "model_report.png"
+            fig.savefig(png_file, bbox_inches='tight', dpi=300)
+
+            # Provide a download button for the PNG file
+            with open(png_file, "rb") as file:
+                st.download_button(
+                    label="Download as PNG",
+                    data=file,
+                    file_name="model_report.png",
+                    mime="image/png"
+                )