fix: update methods

app.py

@@ -11,11 +11,10 @@ from gradio_pdf import PDF
 from backends import (  # convert_zerox,
     SUPPORTED_METHODS,
     SUPPORTED_METHODS_METADATA,
-    convert_gemini,
     convert_gmft,
-    convert_img2table,
     convert_pypdf,
     convert_smoldocling,
+    convert_unstructured,
 )
 from backends.settings import ENABLE_DEBUG_MODE
 from utils import remove_images_from_markdown, trim_pages
@@ -53,10 +52,8 @@ def convert_document(path, method, start_page=0, enabled=True):
             path,
             embed_images=True,
         )
-    elif method == "Gemini (API)":
-        text, debug_image_paths = convert_gemini(path, file_name)
-    elif method == "Img2Table (table-only)":
-        text, debug_image_paths = convert_img2table(path, file_name)
+    elif method == "Unstructured":
+        text, debug_image_paths = convert_unstructured(path, file_name)
     elif method == "GMFT (table-only)":
         text, debug_image_paths = convert_gmft(path, file_name)
     elif method == "PyPDF":
@@ -183,9 +180,7 @@ with gr.Blocks(
             with gr.Row():
                 methods = gr.Dropdown(
                     SUPPORTED_METHODS,
-                    label=(
-                        "Conversion methods " f"(select up-to {MAX_SELECTED_METHODS})"
-                    ),
+                    label=("Conversion methods"),
                     value=SUPPORTED_METHODS[:2],
                     multiselect=True,
                 )

backends/__init__.py

@@ -1,7 +1,5 @@
 # flake8: noqa
-from .gemini import convert_gemini
 from .gmft import convert_gmft
-from .img2table import convert_img2table
 from .pypdf import convert_pypdf
 from .smoldocling import convert_smoldocling
 from .unstructured import convert_unstructured
@@ -9,8 +7,6 @@ from .unstructured import convert_unstructured
 __all__ = [
     "convert_smoldocling",
     "convert_unstructured",
-    "convert_gemini",
-    "convert_img2table",
     "convert_gmft",
     "convert_pypdf",
 ]
@@ -19,9 +15,7 @@ SUPPORTED_METHODS = [
     "SmolDocling",
     "PyMuPDF",
     "PyPDF",
-    "Gemini (API)",
     "Unstructured",
-    "Img2Table (table-only)",
     "GMFT (table-only)",
 ]
 SUPPORTED_METHODS_METADATA = {

requirements.txt

@@ -10,4 +10,4 @@ transformers<5.0.0,>=4.45.2
 pypdf
 docling_core
 opencv-contrib-python
-google-generativeai
+flash_attn