C/hange segment box to work with Uwazi validation

huridocs · Oct 21, 2024 · c3108d4 · c3108d4
1 parent b506e08
commit c3108d4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/app.py b/app.py
@@ -73,7 +73,7 @@ async def labeled_data_post(labeled_data: LabeledData):
 @app.post("/prediction_data")
 async def prediction_data_post(prediction_data: PredictionData):
     predictions_data = json.loads(data_path.read_text()) if exists(data_path) else list()
-    predictions_data.append(prediction_data.dict())
+    predictions_data.append(prediction_data.model_dump())
     data_path.write_text(json.dumps(predictions_data))
     return "prediction data saved"
 

diff --git a/data/SegmentBox.py b/data/SegmentBox.py
@@ -9,7 +9,7 @@ class SegmentBox(BaseModel):
     width: float = 3
     height: float = 4
     page_number: int = 1
-    page_width: int = 600
-    page_height: int = 800
-    text: str = ""
-    type: TokenType = TokenType.TEXT
+    # page_width: int = 600
+    # page_height: int = 800
+    # text: str = ""
+    # type: str = "TEXT"