Update app.py

app.py

@@ -15,7 +15,7 @@ def yolo(im, size=640):
     return Image.fromarray(results.ims[0])
 
 def detect(pdf):
-
+    storelist=[]
     path_to_pdf = pdf.name
     imgs = pdf2image.convert_from_path(path_to_pdf)
     for i in range(len(imgs)):
@@ -24,8 +24,11 @@ def detect(pdf):
         print(result_pred)
         for j in range(len(result_pred[0])):
             if result_pred[0][j][5]==1 and result_pred[0][j][4]>0.3:
-                return "find signature in page " +str(i+1)
-    return "do not find signature in pdf file"
+                storelist.append(i+1)
+    if len(storelist)>0:
+        return "find signature in pdf file page:"+str(storelist)
+    else:
+        return "do not find signature in pdf file"
 
 
 inputs = gr.inputs.Image(type='pil', label="Original Image")