asamasach commited on
Commit
73a5507
·
1 Parent(s): dd8b3f1

Fix numpy array boolean ambiguity in image None check

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -170,8 +170,10 @@ def gradio_inference(image, model_display_name, conf_threshold):
170
  model_key = key
171
  break
172
 
173
- if model_key is None or image is None:
174
  return image
 
 
175
 
176
  session = get_session(model_key)
177
  img_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
@@ -206,7 +208,9 @@ def api_inference(image, model_display_name, conf_threshold):
206
  model_key = key
207
  break
208
 
209
- if model_key is None or image is None:
 
 
210
  return []
211
 
212
  session = get_session(model_key)
 
170
  model_key = key
171
  break
172
 
173
+ if model_key is None:
174
  return image
175
+ if image is None:
176
+ return None
177
 
178
  session = get_session(model_key)
179
  img_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
 
208
  model_key = key
209
  break
210
 
211
+ if model_key is None:
212
+ return []
213
+ if image is None:
214
  return []
215
 
216
  session = get_session(model_key)