Update app.py

app.py

@@ -14,8 +14,8 @@ from docling_core.types.doc import DoclingDocument
 from docling_core.types.doc.document import DocTagsDocument
 import torch
 
-import logging
-logging.basicConfig(level=logging.INFO)
+# import logging
+# logging.basicConfig(level=logging.INFO)
 
 # ---------------------------------------------------------------------------------------
 # API Configuration
@@ -246,7 +246,7 @@ def convert_pdf_to_images(pdf_file, dpi=150, max_size=1600):
 #     return markdown_text
 
 def extract_markdown_from_image(image):
-    start_time = time.time()
+    # start_time = time.time()
     prompt_text = "Convert this page to docling."
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -266,8 +266,8 @@ def extract_markdown_from_image(image):
     doc = DoclingDocument(name="ExtractedDocument")
     doc.load_from_doctags(doctags_doc)
     markdown_text = doc.export_to_markdown()
-    processing_time = time.time() - start_time
-    logging.info(f"Inference took {processing_time:.2f} seconds")
+    # processing_time = time.time() - start_time
+    # logging.info(f"Inference took {processing_time:.2f} seconds")
     return markdown_text
 
 # Streamlit UI