Update app.py

app.py

@@ -10,6 +10,8 @@ from paddleocr import PaddleOCR
 from PIL import Image
 import rarfile
 rarfile.UNRAR_TOOL = "unrar"
+import psutil
+import time
 
 ocr = PaddleOCR(use_angle_cls=True, lang='en', det_model_dir='models/det', rec_model_dir='models/rec', cls_model_dir='models/cls')
 
@@ -112,7 +114,7 @@ def remove_text_dynamic_fill(img_path, output_path):
 
 def process_cbz_cbr(files):
     final_output = tempfile.mkdtemp()
-
+    wait_for_cpu()
     for file_path in files:
         if file_path.endswith(".cbz"):
             with zipfile.ZipFile(file_path, 'r') as archive:
@@ -153,10 +155,21 @@ def convert_cbr_to_cbz(cbr_path):
     return cbz_path
 
 
+def wait_for_cpu(threshold=90, interval=3, timeout=30):
+    start = time.time()
+    while psutil.cpu_percent(interval=1) > threshold:
+        print("High CPU usage detected, waiting...")
+        time.sleep(interval)
+        if time.time() - start > timeout:
+            print("Timed out waiting for CPU to cool down.")
+            break
+
+
 demo = gr.Interface(
     fn=process_cbz_cbr,
     inputs=gr.File(file_types=[".cbz"], file_count="multiple", label="Upload only .cbz Comic Files"),
     outputs=gr.File(label="Download Cleaned Zip"),
+    concurrency_limit=1,
     title="Comic Cleaner from .CBZ",
     description="Upload .cbz comics. The app extracts, cleans (removes text), and gives back a zip of cleaned images."
 )