Update app.py

app.py

@@ -212,7 +212,7 @@ class TableExtractionPipeline():
             text = f'{model.config.id2label[cl.item()]}: {p[cl]:0.2f}'
             ax.text(xmin-20, ymin-50, text, fontsize=10,bbox=dict(facecolor='yellow', alpha=0.5))
         plt.axis('off')
-        c1.pyplot()
+        st.pyplot()
 
 
     def crop_tables(self, pil_img, prob, boxes, delta_xmin, delta_ymin, delta_xmax, delta_ymax):
@@ -265,7 +265,7 @@ class TableExtractionPipeline():
 
 
         plt.axis('on')
-        c2.pyplot()
+        st.pyplot()
         return rows, cols
 
     def sort_table_featuresv2(self, rows:dict, cols:dict):
@@ -413,9 +413,9 @@ class TableExtractionPipeline():
 
         df = self.clean_dataframe(df)
         
-        c3.dataframe(df)
+        st.dataframe(df)
         csv = self.convert_df(df)
-        c3.download_button("Download table", csv, "file.csv", "text/csv", key='download-csv')
+        st.download_button("Download table", csv, "file.csv", "text/csv", key='download-csv')
         
         return df