Update app.py

app.py

@@ -9,6 +9,10 @@ os.system('pip install -q torch==1.10.0+cu111 torchvision==0.11+cu111 -f https:/
 #os.system('pip install -q detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html')
 os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
 
+import detectron2
+from detectron2.utils.logger import setup_logger
+setup_logger()
+
 import gradio as gr
 import re
 import string
@@ -72,7 +76,7 @@ def app_outputs(uploaded_pdf):
     num_images = len(images)
 
     if not msg.startswith("Error with the PDF"):
-    
+
         # Extraction of image data (text and bounding boxes)
         dataset, lines, row_indexes, par_boxes, line_boxes = extraction_data_from_image(images)
         # prepare our data in the format of the model