offthewallace commited on
Commit
3f8f957
·
1 Parent(s): ef265ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,8 +22,8 @@ def detect(pdf):
22
  result_pred=result.pred
23
  for j in range(len(result_pred[0])):
24
  if result_pred[0][j][5]==1 and result_pred[i][j][4]>0.5:
25
- return {'not signature': 0, 'signature': 1}
26
- return {'not signature': 1, 'signature': 0}
27
 
28
 
29
  inputs = gr.inputs.Image(type='pil', label="Original Image")
 
22
  result_pred=result.pred
23
  for j in range(len(result_pred[0])):
24
  if result_pred[0][j][5]==1 and result_pred[i][j][4]>0.5:
25
+ return "find signature in page " +str(i)
26
+ return "do not find signature in pdf file"
27
 
28
 
29
  inputs = gr.inputs.Image(type='pil', label="Original Image")