Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def predict(img):
|
|
| 35 |
|
| 36 |
img = PIL.Image.fromarray(img, "RGB")
|
| 37 |
|
| 38 |
-
pred_dict_2 = model_type.
|
| 39 |
|
| 40 |
img,
|
| 41 |
infer_tfms,
|
|
|
|
| 35 |
|
| 36 |
img = PIL.Image.fromarray(img, "RGB")
|
| 37 |
|
| 38 |
+
pred_dict_2 = model_type.end2end_detect(
|
| 39 |
|
| 40 |
img,
|
| 41 |
infer_tfms,
|