asalhi85 commited on
Commit
649e06f
1 Parent(s): 62ff382

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "Smarathon 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
 
 
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