Update app.py

app.py

@@ -41,7 +41,7 @@ def pdf_to_text(file):
             xref, name, ext, color_space, width, height, bpc, image_data, image_mask, smask_data = img
             # Check if image_data is base64 encoded string
             if isinstance(image_data, str) and re.match(r'^[A-Za-z0-9+/]+[=]{0,2}$', image_data):
-                image_data = base64.b64decode(image_data)
+                image_data = safe_base64_decode(image_data)
             try:
                 image = Image.open(io.BytesIO(image_data))
                 latex_code = image_to_latex(image)
@@ -62,6 +62,7 @@ def pdf_to_text(file):
 
     return output_file_name
 
+
 iface = gr.Interface(fn=pdf_to_text,
                      inputs=gr.inputs.File(label="Your PDF"),
                      outputs=gr.outputs.File(label="Download TXT"),