Upload 4 files

app.py

@@ -0,0 +1,39 @@
+import streamlit as st
+from ultralytics import YOLO
+from PIL import Image
+import tempfile
+ 
+model = YOLO("best.pt")
+ 
+st.title("Circuit Component Detection")
+ 
+uploaded_file = st.file_uploader(
+    "Upload Circuit Image",
+    type=["jpg", "png", "jpeg"]
+)
+ 
+if uploaded_file is not None:
+ 
+    image = Image.open(uploaded_file)
+ 
+    st.image(image, caption="Uploaded Image")
+ 
+    temp_file = tempfile.NamedTemporaryFile(
+        delete=False,
+        suffix=".jpg"
+    )
+ 
+    image.save(temp_file.name)
+ 
+    results = model.predict(
+        source=temp_file.name,
+        conf=0.1,
+        save=False
+    )
+ 
+    annotated = results[0].plot()
+ 
+    st.image(
+        annotated,
+        caption="Detection Result"
+    )

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c54c4e80f543ee94e5a1d5d45417e661a43c968897bcddc6abf812a6acc24fb5
+size 52200722

packages.txt

@@ -0,0 +1 @@
+libgl1

requirements.txt

@@ -0,0 +1,7 @@
+ultralytics
+opencv-python-headless
+pillow
+numpy
+streamlit
+torch
+torchvision