Spaces:

amn23
/

car-image-classification

Sleeping

App Files Files Community

Amina commited on Sep 14, 2025

Commit

d53b194

1 Parent(s): 6f75e04

Add files with LFS tracking

Browse files

Files changed (12) hide show

.idea/.gitignore +8 -0
.idea/car-damage-app.iml +10 -0
.idea/inspectionProfiles/Project_Default.xml +12 -0
.idea/inspectionProfiles/profiles_settings.xml +6 -0
.idea/misc.xml +6 -0
.idea/modules.xml +8 -0
.idea/vcs.xml +6 -0
Dockerfile +13 -0
app.py +39 -0
best.pt +3 -0
index.html +118 -0
requirements.txt +6 -0

.idea/.gitignore ADDED Viewed

	@@ -0,0 +1,8 @@

+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/car-damage-app.iml ADDED Viewed

	@@ -0,0 +1,10 @@

+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/.venv" />
+    </content>
+    <orderEntry type="jdk" jdkName="Python 3.12 (car-damage-app)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml ADDED Viewed

	@@ -0,0 +1,12 @@

+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPep8NamingInspection" enabled="true" level="WEAK WARNING" enabled_by_default="true">
+      <option name="ignoredErrors">
+        <list>
+          <option value="N806" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml ADDED Viewed

	@@ -0,0 +1,6 @@

+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml ADDED Viewed

	@@ -0,0 +1,6 @@

+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.12 (car-damage-app)" />
+  </component>
+</project>

.idea/modules.xml ADDED Viewed

	@@ -0,0 +1,8 @@

+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/car-damage-app.iml" filepath="$PROJECT_DIR$/.idea/car-damage-app.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml ADDED Viewed

	@@ -0,0 +1,6 @@

+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

Dockerfile ADDED Viewed

	@@ -0,0 +1,13 @@

+FROM python:3.9-slim
+WORKDIR /code
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+COPY ./best.pt /code/best.pt
+COPY ./app.py /code/app.py
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py ADDED Viewed

	@@ -0,0 +1,39 @@

+from fastapi import FastAPI, File, UploadFile
+from PIL import Image
+import torch
+import io
+app = FastAPI(title="Car Damage Detection API")
+# Load custom YOLOv8 model.
+model = torch.hub.load('ultralytics/yolov8', 'custom', path='best.pt')
+@app.get("/")
+def read_root():
+    """ A simple endpoint to check if the server is running. """
+    return {"status": "ok", "message": "Car Damage Detection API is running!"}
+@app.post("/detect")
+async def detect_damage(file: UploadFile = File(...)):
+    """
+    This endpoint receives an image, runs inference, and returns detections.
+    """
+    # Read image content from the uploaded file
+    contents = await file.read()
+    image = Image.open(io.BytesIO(contents)).convert("RGB")
+    # Run the model on the image
+    results = model(image)
+    # Extract results into a clean format
+    predictions_df = results.pandas().xyxy[0]
+    # Filter detections by a confidence threshold (e.g., > 50%)
+    detections = predictions_df[predictions_df['confidence'] > 0.5]
+    # Convert the results to a JSON-friendly list of dictionaries
+    output = detections[['name', 'confidence']].to_dict(orient='records')
+    return {"detections": output}

best.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e5b520423c03c31efe05a1b5c9b9387b2a15a8ecb177c47e7d6e3a5bad01fb57
+size 6250474

index.html ADDED Viewed

	@@ -0,0 +1,118 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Car Damage Detector</title>
+    <style>
+        body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif; text-align: center; margin: 0; background-color: #f0f2f5; }
+        .container { max-width: 600px; margin: 50px auto; padding: 30px; background-color: #fff; box-shadow: 0 4px 12px rgba(0,0,0,0.1); border-radius: 8px; }
+        h1 { color: #333; }
+        #image-preview { display: none; max-width: 100%; height: auto; margin-top: 20px; border-radius: 4px; border: 1px solid #ddd; }
+        .upload-area { border: 2px dashed #007bff; border-radius: 8px; padding: 40px; margin: 20px 0; cursor: pointer; background-color: #f8f9fa; }
+        .upload-area p { margin: 0; color: #555; }
+        button { font-size: 1.1em; padding: 12px 25px; cursor: pointer; background-color: #007bff; color: white; border: none; border-radius: 5px; transition: background-color 0.3s; }
+        button:hover { background-color: #0056b3; }
+        button:disabled { background-color: #aaa; cursor: not-allowed; }
+        #results { margin-top: 30px; font-size: 1.1em; text-align: left; }
+        #results ul { list-style-type: none; padding: 0; }
+        #results li { background-color: #e9ecef; padding: 10px; margin-bottom: 8px; border-radius: 4px; }
+        #results .confidence { color: #28a745; font-weight: bold; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>🚗 Car Damage Detector</h1>
+        <p>Upload a photo to automatically check for scratches, dents, or rust.</p>
+        <div class="upload-area" id="upload-area">
+            <p>Click here or drag & drop an image</p>
+        </div>
+        <input type="file" id="image-upload" accept="image/*" hidden>
+        <button id="check-button" disabled>Check for Damage</button>
+        <img id="image-preview" src="" alt="Your uploaded image">
+        <div id="results"></div>
+    </div>
+    <script>
+        // ⚠️ IMPORTANT: REPLACE THIS URL with your actual Hugging Face Space URL!
+        const API_URL = "https://YOUR_USERNAME-YOUR_SPACE_NAME.hf.space/detect";
+        const uploadArea = document.getElementById('upload-area');
+        const imageUpload = document.getElementById('image-upload');
+        const checkButton = document.getElementById('check-button');
+        const imagePreview = document.getElementById('image-preview');
+        const resultsDiv = document.getElementById('results');
+        let selectedFile = null;
+        uploadArea.addEventListener('click', () => imageUpload.click());
+        imageUpload.addEventListener('change', handleFileSelect);
+        function handleFileSelect(event) {
+            selectedFile = event.target.files[0];
+            if (selectedFile) {
+                imagePreview.src = URL.createObjectURL(selectedFile);
+                imagePreview.style.display = 'block';
+                checkButton.disabled = false;
+                resultsDiv.innerHTML = '';
+            }
+        }
+        checkButton.addEventListener('click', async () => {
+            if (!selectedFile) {
+                alert("Please select an image first!");
+                return;
+            }
+            resultsDiv.innerHTML = "<p>Analyzing, please wait... 🧐</p>";
+            checkButton.disabled = true;
+            checkButton.innerText = "Analyzing...";
+            const formData = new FormData();
+            formData.append('file', selectedFile);
+            try {
+                const response = await fetch(API_URL, {
+                    method: 'POST',
+                    body: formData,
+                });
+                if (!response.ok) {
+                    throw new Error(`Server error: ${response.statusText}`);
+                }
+                const data = await response.json();
+                displayResults(data.detections);
+            } catch (error) {
+                resultsDiv.innerHTML = `<p style="color: red;">An error occurred: ${error.message}</p>`;
+                console.error("Error:", error);
+            } finally {
+                checkButton.disabled = false;
+                checkButton.innerText = "Check for Damage";
+            }
+        });
+        function displayResults(detections) {
+            if (detections.length === 0) {
+                resultsDiv.innerHTML = "<p>✅ No damage detected!</p>";
+                return;
+            }
+            let html = "<h3>Detected Issues:</h3><ul>";
+            detections.forEach(det => {
+                const confidence = (det.confidence * 100).toFixed(1);
+                html += `<li><strong>${det.name}</strong> <span class="confidence">(${confidence}%)</span></li>`;
+            });
+            html += "</ul>";
+            resultsDiv.innerHTML = html;
+        }
+    </script>
+</body>
+</html>

requirements.txt ADDED Viewed

	@@ -0,0 +1,6 @@

+fastapi
+uvicorn
+python-multipart
+torch
+ultralytics
+Pillow