Fix numpy array boolean ambiguity in image None check
Browse files
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
|
| 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
|
|
|
|
|
|
|
| 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)
|