Update app.py

app.py

@@ -25,7 +25,9 @@ def model():
     
     for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
         box = [round(i, 2) for i in box.tolist()]
-        st.write("Prediction:", results[0]['label'], "| Score:", results[0]['score'])
+        st.write(f"Detected {model.config.id2label[label.item()]} with confidence ")
+        st.write(f"{round(score.item(), 3)} at location {box}")
+        #st.write("Prediction:", results[0]['label'], "| Score:", results[0]['score'])
         #print(
          #   f"Detected {model.config.id2label[label.item()]} with confidence "
          #   f"{round(score.item(), 3)} at location {box}"