Update app.py

app.py

@@ -434,10 +434,10 @@ class TableExtractionPipeline():
         '''
         image = Image.open(image_path).convert("RGB")
         model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=TD_THRESHOLD)
-        st.write(bboxes_scaled)
-        #if bboxes_scaled.nelement() == 0:
-            #st.write('No table found in the pdf-page image')
-            #return ''
+
+        if bboxes_scaled.nelement() == 0:
+            st.write('No table found in the pdf-page image')
+            return ''
         
         # try:
         # st.write('Document: '+image_path.split('/')[-1])