fixed ocr

app/main.py

@@ -100,10 +100,10 @@ async def perform_ocr(
     serialized_result = [
         {
             "bbox": record.bbox,
-            "confidence": record.confidence,
+            # "confidence": record.confidence,
             "text": record.prediction,
             "cuts": record.cuts,
-            "line_id": record.line_id,
+            # "line_id": record.line_id,
         }
         for record in result
     ]
@@ -179,13 +179,14 @@ async def process_all(
         raise HTTPException(status_code=400, detail=f"Model '{model_name}' not found or invalid")
     
     result = list(rpred.rpred(model, bw_img, segmentation))
+    print(result)
     serialized_result = [
         {
             "bbox": record.bbox,
-            "confidence": record.confidence,
+            # "confidence": record.confidence,
             "text": record.prediction,
             "cuts": record.cuts,
-            "line_id": record.line_id,
+            # "line_id": record.line_id,
         }
         for record in result
     ]

send_image.py

@@ -2,7 +2,8 @@ import requests
 import os
 
 # API endpoint
-url = "https://wjbmattingly-kraken-api.hf.space/ocr"
+# url = "https://wjbmattingly-kraken-api.hf.space/ocr"
+url = "http://127.0.0.1:8000/process_all"
 
 # Path to the image file
 image_path = os.path.join("data", "ms.jpg")
@@ -26,9 +27,9 @@ if response.status_code == 200:
     
     for record in ocr_results:
         print(f"Text: {record['text']}")
-        print(f"Confidence: {record['confidence']}")
+        # print(f"Confidence: {record['confidence']}")
         print(f"Bounding Box: {record['bbox']}")
-        print()
+        # print()
     
 else:
     print(f"Error: {response.status_code}")