Update app.py

app.py

@@ -52,7 +52,7 @@ model.to(device);
 def app_outputs(uploaded_pdf):
     filename, msg, images = pdf_to_images(uploaded_pdf)
     num_images = len(images)
-
+    path = os.path.dirname(__file__)
     if not msg.startswith("Error with the PDF"):
     
         # Extraction of image data (text and bounding boxes)
@@ -67,10 +67,12 @@ def app_outputs(uploaded_pdf):
         # Get labeled images with lines bounding boxes
         images = get_labeled_images(dataset, images_ids_list, bboxes_list_dict, probs_dict_dict)
 
+        
+      
         img_files = list()
         # get image of PDF without bounding boxes
         for i in range(num_images):
-            if filename != "files/blank.png": img_file = f"img_{i}_" + filename.replace(".pdf", ".png")
+            if filename != path+"/blank.png": img_file = f"img_{i}_" + filename.replace(".pdf", ".png")
             else: img_file = filename.replace(".pdf", ".png")
             images[i].save(img_file)
             img_files.append(img_file)