Update app2.py

app2.py

@@ -11,6 +11,7 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from botocore.exceptions import NoCredentialsError
+from ultralytics import YOLO
 
 # ==============================================================================
 # 1. CONFIGURAÇÃO E CLIENTES (CARREGADOS NA INICIALIZAÇÃO)
@@ -110,6 +111,23 @@ def process_image_bytes(img_bytes: bytes):
     
     return analysis_results, corrosion_vis_rgb
 
+def segmentate():
+
+    # Predição basead no melhor modelo gerado (best.pt)
+    modelo = YOLO('/content/resultado/Parafuso3/weights/best.pt')
+    resultados = modelo.predict(
+    source='/content/dados/images/train',
+    imgsz=1280,
+    conf=0.7,
+    iou=0.3,
+    save=True,
+    save_crop=True,
+    project='/content/resultado'
+)
+
+    
+    
+
 # ==============================================================================
 # 3. ENDPOINTS DA API
 # ==============================================================================
@@ -118,6 +136,7 @@ def process_image_bytes(img_bytes: bytes):
 def read_root():
     return {"status": "ok", "message": "API de Análise de Corrosão no ar!"}
 
+
 @app.post("/analyze")
 async def analyze(file: UploadFile = File(...)):
     """