xinyu1205 commited on
Commit
b8780b2
1 Parent(s): 4ff9edd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def inference(raw_image, input_tag):
31
  raw_image = raw_image.resize((image_size, image_size))
32
 
33
  image = transform(raw_image).unsqueeze(0).to(device)
34
- model.threshold = 0.69
35
  if input_tag == '' or input_tag == 'none' or input_tag == 'None':
36
  input_tag_list = None
37
  else:
 
31
  raw_image = raw_image.resize((image_size, image_size))
32
 
33
  image = transform(raw_image).unsqueeze(0).to(device)
34
+ model.threshold = 0.68
35
  if input_tag == '' or input_tag == 'none' or input_tag == 'None':
36
  input_tag_list = None
37
  else: