Update app.py

app.py

@@ -63,6 +63,19 @@ async def segment_image(file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
+@app.get("/models")
+def list_models():
+    return {
+        "models": [
+            {
+                "name": "sam-cvat",
+                "type": "segmentation",
+                "labels": ["object"]
+            }
+        ]
+    }        
+
 # CVAT-specific endpoint
 @app.post("/predict")
 async def predict_for_cvat(body: str = Form(...)):
@@ -99,15 +112,17 @@ async def predict_for_cvat(body: str = Form(...)):
         # Convert mask to CVAT format
         height, width = mask.shape
         rle = mask_to_rle(mask)
-        
         return {
+            "model": "sam-cvat",
             "annotations": [{
-                "ObjectID": 1,
-                "ObjectScore": float(scores[best_mask_idx]),
-                "RLE": rle,
-                "PredictionType": "mask",
-                "width": width,
-                "height": height
+                "name": "object",
+                "score": float(scores[best_mask_idx]),
+                "mask": {
+                    "rle": rle,
+                    "width": width,
+                    "height": height
+                },
+                "type": "mask"
             }]
         }
     except Exception as e: