update

utils.py

@@ -79,16 +79,24 @@ def infer_uploaded_image(conf, model):
             )
 
     if source_img:
-        if st.button("Execution"):
+        if st.button("Detect"):
             with st.spinner("Running..."):
                 res = model.predict(uploaded_image,
                                     conf=conf)
                 boxes = res[0].boxes
+                dclasses = boxes.cls.numpy(force=True)
+                names = res[0].names
+
+                dnames = []
+                for cid in dclasses:
+                    dnames.append(names[cid])
+
+                #print('names', names)
                 res_plotted = res[0].plot()[:, :, ::-1]
 
                 with col2:
                     st.image(res_plotted,
-                             caption="Detected Image",
+                             caption="Detected:{}".format(','.join(dnames)),
                              use_column_width=True)
                     try:
                         with st.expander("Detection Results"):
@@ -114,7 +122,7 @@ def infer_uploaded_video(conf, model):
         st.video(source_video)
 
     if source_video:
-        if st.button("Execution"):
+        if st.button("Detect"):
             with st.spinner("Running..."):
                 try:
                     tfile = tempfile.NamedTemporaryFile()