Update app.py

app.py

@@ -176,9 +176,10 @@ if uploaded_files:
                         # Table Extraction
                         # L = []
                         output_list = []
-                        st.write(pdf.name)
-                        st.write(type(pdf.name))
-                        elements = partition_pdf(filename=pdf.name, strategy=strategy, infer_table_structure=True, model_name=model_name)
+                        try:
+                            elements = partition_pdf(filename=pdf.name, strategy=strategy, infer_table_structure=True, model_name=model_name)
+                        except:
+                            elements = partition_pdf(filename=pdf.name, strategy=strategy, infer_table_structure=True)
                         with NamedTemporaryFile(dir=".", suffix=".json") as f:
                             elements_to_json(elements, filename=f"{f.name.split('/')[-1]}")
                             json_file_path = os.path.abspath(f.name)  # Get the absolute file path