Initial commit for garbage redressal space

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install OS dependencies
+RUN apt-get update && apt-get install -y git ffmpeg libsm6 libxext6 && rm -rf /var/lib/apt/lists/*
+
+# Copy files
+COPY app ./app
+COPY app/requirements.txt .
+
+# Install Python dependencies
+RUN pip install --upgrade pip && pip install -r requirements.txt
+
+# Set the entrypoint
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/best.onnx

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

app/best.pt

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

app/main.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI, UploadFile, File
+from ultralytics import YOLO
+import uvicorn
+from PIL import Image
+import io
+
+app = FastAPI()
+model = YOLO("app/model.pt")  # Load the trained YOLOv11/12 model
+
+@app.get("/")
+def read_root():
+    return {"message": "Welcome to the Garbage Detection API"}
+
+@app.post("/predict/")
+async def predict(file: UploadFile = File(...)):
+    image_bytes = await file.read()
+    image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+
+    results = model(image)
+    boxes = results[0].boxes.xyxy.tolist()  # List of [x1, y1, x2, y2]
+    confidences = results[0].boxes.conf.tolist()
+
+    return {"boxes": boxes, "confidences": confidences}

app/requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+pillow
+ultralytics