Update app.py

app.py

@@ -3,6 +3,8 @@ import re, json, tempfile
 import cv2
 import numpy as np
 import gradio as gr
+import gc
+import psutil
 from ultralytics import YOLO
 from paddleocr import PaddleOCR
 
@@ -27,6 +29,10 @@ ocr = PaddleOCR(
 )
 # Force our exact char map (no dict file needed)
 ocr.text_recognizer.character = CHAR_LIST
+def print_mem_usage(tag=""):
+    mem = psutil.virtual_memory()
+    print(f"[{tag}] RAM usage: {mem.used / 1024**2:.2f} MB / {mem.total / 1024**2:.2f} MB ({mem.percent}%)")
+
 
 # ─── 4) Normalize & format OCR output
 def normalize_ocr(recs):
@@ -94,6 +100,9 @@ def run_image(img, conf=0.25):
 
         try:
             recs = ocr.ocr(warped, det=False, cls=True)
+            # After OCR call:
+            gc.collect()
+            print_mem_usage("After OCR")
         except:
             recs = []
         txt, score = normalize_ocr(recs)
@@ -127,6 +136,9 @@ def run_video(video_file, conf=0.25):
 
             try:
                 recs = ocr.ocr(warped, det=False, cls=True)
+                 # After OCR call:
+                gc.collect()
+                print_mem_usage("After OCR")
             except:
                 recs = []
             txt, score = normalize_ocr(recs)