Upload 3 files

2.py

@@ -0,0 +1,59 @@
+import streamlit as st
+from ultralytics import YOLO
+from PIL import Image
+import tempfile
+import os
+
+# import pkg_resources
+
+# def generate_requirements(filename="requirements.txt"):
+#     """
+#     Generate a requirements.txt file containing installed pip packages and versions.
+#     """
+#     # Get all installed packages
+#     installed_packages = pkg_resources.working_set
+#     packages_list = sorted([f"{pkg.key}=={pkg.version}" for pkg in installed_packages])
+
+#     # Write to requirements.txt
+#     with open(filename, "w") as f:
+#         for package in packages_list:
+#             f.write(package + "\n")
+    
+#     print(f"Requirements saved to {filename}")
+
+# if __name__ == "__main__":
+#     generate_requirements()
+
+
+# Load YOLO model (COCO-pretrained)
+model = YOLO('yolov8s.pt')  # or 'yolov10s.pt'
+
+st.title("YOLO Object Detection with Streamlit")
+st.write("Upload an image and see object detection results in real-time!")
+
+# Upload image
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    # Convert uploaded file to PIL image
+    image = Image.open(uploaded_file).convert("RGB")
+    st.image(image, caption="Uploaded Image", width='stretch')
+
+    # Temporary save to disk for YOLO inference
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
+        temp_path = tmp_file.name
+        image.save(temp_path)
+
+    # Run YOLO inference
+    results = model(temp_path)
+
+    # Show detection results in Streamlit
+    st.image(results[0].plot(), caption="Detected Objects", width='stretch')
+
+    # # Save results to output folder
+    # output_dir = "outputs"
+    # os.makedirs(output_dir, exist_ok=True)
+    # results.save(output_dir)
+    # st.write(f"Detection results saved in `{output_dir}/`")
+    # Clean up temporary file
+    os.remove(temp_path)   

Dockerfile

@@ -0,0 +1,19 @@
+# Base image
+FROM python:3.11-slim
+
+# Set working directory
+WORKDIR /app
+
+# Copy requirements and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy project files
+COPY . .
+
+# Expose port
+EXPOSE 8000
+
+# Default command
+CMD ["python", "2.py"]

requirements.txt

@@ -0,0 +1,2 @@
+streamlit==1.50.0
+ultralytics==8.3.220