saritha5 commited on
Commit
1379f3f
1 Parent(s): f5803bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -17,6 +17,11 @@ model = core.Model.load(MODEL_PATH, ['cross_arm','pole','tag'])
17
  #warnings.warn(msg)
18
 
19
  st.title("Object Detection")
 
 
 
 
 
20
 
21
 
22
  #def detect_object(IMAGE_PATH):
 
17
  #warnings.warn(msg)
18
 
19
  st.title("Object Detection")
20
+ image = utils.read_image(IMAGE_PATH)
21
+ predictions = model.predict(image)
22
+ labels, boxes, scores = predictions
23
+
24
+ plt.show(IMAGE_PATH)
25
 
26
 
27
  #def detect_object(IMAGE_PATH):