Update app.py

app.py

@@ -53,7 +53,8 @@ def detect_object(IMAGE_PATH):
     filtered_labels = [labels[i] for i in num_list]
     show_labeled_image(image, filtered_boxes, filtered_labels)
     
-    show_image(image,filtered_boxes,filtered_labels)
+    fig1 = show_image(image,filtered_boxes,filtered_labels)
+    st.image(fig1)
     #img_array = img_to_array(img)
 def show_image(image, boxes, labels=None):
     """Show the image along with the specified boxes around detected objects.
@@ -107,6 +108,7 @@ def show_image(image, boxes, labels=None):
     cp = os.path.abspath(os.getcwd()) + '/foo.png'
     plt.savefig(cp)
     plt.close(fig)
+    return cp
     #print(type(plt
 
 file = st.file_uploader('Upload an Image',type=(["jpeg","jpg","png"]))