rudimentary load controll

localization.py

@@ -7,6 +7,8 @@ from jax import numpy as jnp
 import pandas as pd
 import numpy as np
 import requests
+import psutil
+import time
 import jax
 import gc
 
@@ -138,6 +140,13 @@ def app():
 
     if compute:
 
+        with st.spinner('Waiting for resources...'):
+            sleep_time = 5
+            print('CPU_load', psutil.cpu_percent())
+            while psutil.cpu_percent() > 90:
+                time.sleep(sleep_time)
+
+
         if not caption or not image_url:
             st.error("Please choose one image and at least one label")
         else:

requirements.txt

@@ -6,4 +6,5 @@ torchvision
 natsort
 stqdm
 pandas
-requests
+requests
+psutil