Update app.py

app.py

@@ -102,6 +102,9 @@ def remove_text_dynamic_fill(img_path, output_path):
 
 def process_folder(input_files):
     temp_output = tempfile.mkdtemp()
+    
+
+    wait_for_cpu();    
     for file in input_files:
         filename = os.path.basename(file.name)
         output_path = os.path.join(temp_output, filename)
@@ -110,10 +113,23 @@ def process_folder(input_files):
     zip_path = shutil.make_archive(temp_output, 'zip', temp_output)
     return zip_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_folder,
     inputs=gr.File(file_types=[".jpg", ".jpeg", ".png"], file_count="multiple", label="Upload Comic Images"),
     outputs=gr.File(label="Download Cleaned Zip"),
+    concurrency_limit=1,
     title="Comic Text Cleaner",
     description="Upload comic images and get a zip of cleaned versions (text removed). Uses PaddleOCR for detection."
 )