valentynliubchenko commited on
Commit
6709c36
·
1 Parent(s): 2b9ca25

fixed input parameters

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -20,10 +20,12 @@ def process_image(input_image, model_name, conf):
20
  print(f"conf : {conf}")
21
 
22
  if model_name is None:
23
- model_name = model_names[0]
 
24
 
25
  if conf is None:
26
- conf = 0.6
 
27
 
28
  model_index = model_names.index(model_name)
29
  print('model_index: ')
 
20
  print(f"conf : {conf}")
21
 
22
  if model_name is None:
23
+ return None, "No objects detected."
24
+ # model_name = model_names[0]
25
 
26
  if conf is None:
27
+ return None, "No objects detected."
28
+ # conf = 0.6
29
 
30
  model_index = model_names.index(model_name)
31
  print('model_index: ')