Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 from fastapi import FastAPI, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from ultralytics import YOLO
@@ -9,7 +8,6 @@ import uvicorn
 
 app = FastAPI()
 
-# CORS 허용
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -18,58 +16,42 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# ===========================================
-#  YOLOv8 segmentation model (from keremberke)
-# ===========================================
+print("🔵 Loading model: best.pt")
 model = YOLO("best.pt")
 print("✅ Model loaded")
 
 @app.post("/predict")
 async def predict(img: UploadFile = File(...)):
     try:
-        # 이미지 로딩
         bytes_data = await img.read()
         image = Image.open(io.BytesIO(bytes_data)).convert("RGB")
         np_img = np.array(image)
 
-        # YOLO inference
         results = model(np_img)
         result = results[0]
 
-        # segmentation mask 여부로 crack 유무 판단
         has_mask = result.masks is not None
 
         if not has_mask:
             return {
-                "data": [
-                    {"label": "normal", "confidence": 1.0}
-                ]
+                "data": [{"label": "normal", "confidence": 1.0}]
             }
 
-        # 가장 높은 confidence 추출
         if result.boxes is not None and len(result.boxes) > 0:
             conf = float(result.boxes.conf.max().item())
         else:
-            conf = 0.85  # 기본값
+            conf = 0.85  
 
         return {
-            "data": [
-                {
-                    "label": "crack",
-                    "confidence": conf
-                }
-            ]
+            "data": [{"label": "crack", "confidence": conf}]
         }
 
     except Exception as e:
         print("❌ Prediction error:", e)
         return {
-            "data": [
-                {"label": "normal", "confidence": 1.0}
-            ],
+            "data": [{"label": "normal", "confidence": 1.0}],
             "error": str(e)
         }
 
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)