Update app.py

app.py

@@ -28,21 +28,29 @@ with st.sidebar:
 
 
 
-#def detect_object(IMAGE_PATH):
- #   image = utils.read_image(IMAGE_PATH) 
+def detect_object(IMAGE_PATH):
+    image = utils.read_image(IMAGE_PATH) 
   #  predictions = model.predict(image)
    # labels, boxes, scores = predictions
 
 
-    #thresh=0.2
-    #filtered_indices=np.where(scores>thresh)
-    #filtered_scores=scores[filtered_indices]
-    #filtered_boxes=boxes[filtered_indices]
-    #num_list = filtered_indices[0].tolist()
-    #filtered_labels = [labels[i] for i in num_list]
-    #visualize.show_labeled_image(image, filtered_boxes, filtered_labels)
+    thresh=0.2
+    filtered_indices=np.where(scores>thresh)
+    filtered_scores=scores[filtered_indices]
+    filtered_boxes=boxes[filtered_indices]
+    num_list = filtered_indices[0].tolist()
+    filtered_labels = [labels[i] for i in num_list]
+    st.show_labeled_image(image, filtered_boxes, filtered_labels)
     #img_array = img_to_array(img)
 
+file = st.file_uploader('Upload an Image',type=(["jpeg","jpg","png"]))
+
+if file is None:
+    st.write("Please upload an image file")
+else:
+    image= Image.open(file)
+    st.image(image,use_column_width = True)
+    detect_object(file)