Update app.py

app.py

@@ -22,7 +22,25 @@ def save_image(upload_image_path):
         ('Browse', 'Camera'))
 
     if genre == 'Camera':
-        uploaded_image = st.camera_input("Take a picture")
+        #uploaded_image = st.camera_input("Take a picture")
+        clear_detect_directory()
+
+        command = [
+            "python",
+            "yolov5/detect.py",
+            "--weights", "best.pt",
+            "--img", "416",
+            "--conf", "0.66",
+            "--source", upload_image_path
+        ]
+
+        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        std_out, std_err = process.communicate()
+        if process.returncode != 0:
+            error_message = f"Error: {std_err}"
+            st.text(error_message)
+        else:
+            st.text(std_out)
     else:
         uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])