ozyman commited on
Commit
fc6a3c7
·
1 Parent(s): 73cb092

added conf to file name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ os.environ['OMP_NUM_THREADS'] = '4'
32
  os.environ['AWS_ACCESS_KEY_ID'] = 'AKIA3JAMX4K53MFDKMGJ'
33
  os.environ['AWS_SECRET_ACCESS_KEY'] = 'lHf9xIwdgO3eXrE9a4KL+BTJ7af2cgZJYRRxw4NI'
34
 
35
- app_version = 'ddn1'
36
 
37
  device = torch.device("cpu")
38
  labels = ['Live', 'Spoof']
@@ -190,7 +190,7 @@ def dsdg_model_inference(img, bbox, dsdg_thresh):
190
  cv.putText(img_dsdg, text, (x, y2 + 30),
191
  cv.FONT_HERSHEY_COMPLEX, 1, color_dsdg)
192
  cls_dsdg = 1 if cls_dsdg == 'Real' else 0
193
- return img_dsdg, confidences_dsdg, cls_dsdg
194
 
195
 
196
  def inference(img, dsdg_thresh):
 
32
  os.environ['AWS_ACCESS_KEY_ID'] = 'AKIA3JAMX4K53MFDKMGJ'
33
  os.environ['AWS_SECRET_ACCESS_KEY'] = 'lHf9xIwdgO3eXrE9a4KL+BTJ7af2cgZJYRRxw4NI'
34
 
35
+ app_version = 'ddn2'
36
 
37
  device = torch.device("cpu")
38
  labels = ['Live', 'Spoof']
 
190
  cv.putText(img_dsdg, text, (x, y2 + 30),
191
  cv.FONT_HERSHEY_COMPLEX, 1, color_dsdg)
192
  cls_dsdg = 1 if cls_dsdg == 'Real' else 0
193
+ return img_dsdg, confidences_dsdg, res_dsdg * 100
194
 
195
 
196
  def inference(img, dsdg_thresh):