Update app.py

app.py

@@ -13,17 +13,27 @@ def yolo(im, size=640):
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.ims[0])
 
-def detect(image):
-    results = model(image)  
-    results.render() 
-    return Image.fromarray(results.ims[0])
+def detect(pdf):
+
+    path_to_pdf = pdf.name
+    imgs = pdf2image.convert_from_path(path_to_pdf)
+    for i in range(len(imgs):
+        
+        result = model(np.array(imgs[i]))
+        result_pred=result.pred
+        
+        for j in range(len(result_pred[0])):
+            if result_pred[0][j][5]==1 and result_pred[i][j][4]>0.5:
+                return {'not signature': 0, 'signature': 1}
+    return return {'not signature': 1, 'signature': 0}
 
 
 inputs = gr.inputs.Image(type='pil', label="Original Image")
-outputs = gr.outputs.Image(type="pil", label="Output Image")
+#outputs = gr.outputs.Image(type="pil", label="Output Image")
 
-title = "Object detection from Infrared image using YOLOv5n"
+title = "Object detection for signature detections in pdf file "
+description = "Drop a PDF (WARNING: only the first page will be converted into an image)."
 
 
-gr.Interface(detect, inputs, outputs, title=title, theme="huggingface").launch(debug=True)
+gr.Interface(detect, inputs=gr.File(label="PDF"), outputs="label", title=title, description=description,theme="huggingface").launch(debug=True)