Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def inference(image_url, image, min_score):
|
|
| 50 |
im = cv2.imdecode(im, cv2.IMREAD_COLOR)
|
| 51 |
else:
|
| 52 |
# Model expect BGR!
|
| 53 |
-
im = image[:,:,::-1]
|
| 54 |
|
| 55 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
|
| 56 |
predictor = DefaultPredictor(cfg)
|
|
@@ -63,7 +63,7 @@ def inference(image_url, image, min_score):
|
|
| 63 |
return out.get_image()
|
| 64 |
|
| 65 |
|
| 66 |
-
title = "
|
| 67 |
description = "This demo introduces an interactive playground for our trained Detectron2 model."
|
| 68 |
article = '<p>Detectron model is available from our repository <a href="https://github.com/asalhi/Smartathon-Baseer">here</a>.</p>'
|
| 69 |
|
|
|
|
| 50 |
im = cv2.imdecode(im, cv2.IMREAD_COLOR)
|
| 51 |
else:
|
| 52 |
# Model expect BGR!
|
| 53 |
+
im = image#[:,:,::-1]
|
| 54 |
|
| 55 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
|
| 56 |
predictor = DefaultPredictor(cfg)
|
|
|
|
| 63 |
return out.get_image()
|
| 64 |
|
| 65 |
|
| 66 |
+
title = "Smartathon Phase2 Demo - Baseer"
|
| 67 |
description = "This demo introduces an interactive playground for our trained Detectron2 model."
|
| 68 |
article = '<p>Detectron model is available from our repository <a href="https://github.com/asalhi/Smartathon-Baseer">here</a>.</p>'
|
| 69 |
|