Update app.py

app.py

@@ -1,25 +1,31 @@
 import gradio as gr
 from docling.document_converter import DocumentConverter
 import spaces
+from marker.converters.pdf import PdfConverter
+from marker.models import create_model_dict
+from marker.config.parser import ConfigParser
 
+# Docling
+docling_converter = DocumentConverter()
+
+# Marker
+config_parser = ConfigParser({
+    "output_format": "json"
+})
+marker_converter = PdfConverter(
+    config=config_parser.generate_config_dict(),
+    artifact_dict=create_model_dict(),
+    processor_list=config_parser.get_processors(),
+    renderer=config_parser.get_renderer()
+)
 
 def convert_document(file, method):
     if method == "Docling":
-        # Load document and convert it using Docling
-        converter = DocumentConverter()
-        result = converter.convert(file.name)
-    
-        # Check available attributes in DoclingDocument
-        available_attributes = dir(result.document)
-        document = result.document
-    
-    
-        # Output
-        converted_text = result.document.export_to_markdown()
-    
-        return converted_text
+        result = docling_converter.convert(file.name)
+        
+        return result.document.export_to_markdown()
     elif method == "Marker":
-        return 'unsupported method'
+        return marker_converter(file.name)
     else:
         return 'unknown method'