update model file and improved accuracy

app.py

@@ -9,6 +9,7 @@ import io
 import os
 import requests
 import gdown
+from skimage import io
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -16,23 +17,29 @@ cfg = None
 # Google Drive file URL
 GDRIVE_MODEL_URL = "https://drive.google.com/uc?id=18aEDo-kWOBhg8mAhnbpFkuM6bmmrBH4E"  # Replace 'your-file-id' with the actual file ID from Google Drive
 LOCAL_MODEL_PATH = "model_final.pth"
+
+
 def download_file_from_google_drive(id, destination):
     gdown.download(GDRIVE_MODEL_URL, LOCAL_MODEL_PATH, quiet=False)
 
-                
-file_id = '18aEDo-kWOBhg8mAhnbpFkuM6bmmrBH4E'
-destination = 'model_final.pth'
+
+file_id = "18aEDo-kWOBhg8mAhnbpFkuM6bmmrBH4E"
+destination = "model_final.pth"
 download_file_from_google_drive(file_id, destination)
 
+
 # Download model from Google Drive if not already present locally
 def download_model():
     if not os.path.exists(LOCAL_MODEL_PATH):
         response = requests.get(GDRIVE_MODEL_URL, stream=True)
         if response.status_code == 200:
-            with open(LOCAL_MODEL_PATH, 'wb') as f:
+            with open(LOCAL_MODEL_PATH, "wb") as f:
                 f.write(response.content)
         else:
-            raise Exception(f"Failed to download model from Google Drive: {response.status_code}")
+            raise Exception(
+                f"Failed to download model from Google Drive: {response.status_code}"
+            )
+
 
 # Configuration and model setup
 def setup_model(model_path):
@@ -44,36 +51,39 @@ def setup_model(model_path):
     cfg.MODEL.DEVICE = "cpu"  # Use "cuda" for GPU
     return DefaultPredictor(cfg)
 
+
 # Ensure model is available
 predictor = setup_model(LOCAL_MODEL_PATH)
 
 # Define expected parts and costs
-expected_parts = ['headlamp', 'rear_bumper', 'door', 'hood', 'front_bumper']
+expected_parts = ["headlamp", "rear_bumper", "door", "hood", "front_bumper"]
 cost_dict = {
-    'headlamp': 300,
-    'rear_bumper': 250,
-    'door': 200,
-    'hood': 220,
-    'front_bumper': 250,
-    'other': 150
+    "headlamp": 300,
+    "rear_bumper": 250,
+    "door": 200,
+    "hood": 220,
+    "front_bumper": 250,
+    "other": 150,
 }
 
-@app.route('/')
+
+@app.route("/")
 def home():
-    return render_template('index.html')
+    return render_template("index.html")
 
-@app.route('/upload', methods=['POST'])
+
+@app.route("/upload", methods=["POST"])
 def upload():
-    if 'file' not in request.files:
+    if "file" not in request.files:
         return jsonify({"error": "No file uploaded"}), 400
 
-    file = request.files['file']
-    if file.filename == '':
+    file = request.files["file"]
+    if file.filename == "":
         return jsonify({"error": "No file selected"}), 400
 
     # Load image
-    image = Image.open(file).convert("RGB")
-    image_np = np.array(image)
+    image = io.imread(file)
+    image_np = image
 
     # Run model prediction
     outputs = predictor(image_np)
@@ -86,24 +96,21 @@ def upload():
 
     for j in range(len(instances)):
         class_id = instances.pred_classes[j].item()
-        damaged_part = class_names[class_id] if class_id < len(class_names) else 'unknown'
+        damaged_part = (
+            class_names[class_id] if class_id < len(class_names) else "unknown"
+        )
         if damaged_part not in expected_parts:
-            damaged_part = 'other'
+            damaged_part = "other"
 
-        repair_cost = cost_dict.get(damaged_part, cost_dict['other'])
+        repair_cost = cost_dict.get(damaged_part, cost_dict["other"])
         total_cost += repair_cost
 
-        damage_details.append({
-            'part': damaged_part,
-            'cost_usd': repair_cost
-        })
+        damage_details.append({"part": damaged_part, "cost_usd": repair_cost})
 
-    response = {
-        "damages": damage_details,
-        "total_cost": total_cost
-    }
+    response = {"damages": damage_details, "total_cost": total_cost}
 
     return jsonify(response)
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)