check memory

app.py

@@ -63,4 +63,5 @@ with gr.Blocks(css='style.css') as demo:
             create_demo_draw(model.process_draw)
 
 # demo.queue(api_open=False).launch(server_name='0.0.0.0')
-demo.queue(show_api=False, enable_queue=False).launch(server_name='0.0.0.0')
+# demo.queue(show_api=False, enable_queue=False).launch(server_name='0.0.0.0')
+demo.queue().launch(debug=True) 

demo/demos.py

@@ -1,9 +1,18 @@
 import gradio as gr
 import numpy as np
+import psutil
 
 def create_map():
     return np.zeros(shape=(512, 1024), dtype=np.uint8)+255
 
+def get_system_memory():
+    memory = psutil.virtual_memory()
+    memory_percent = memory.percent
+    memory_used = memory.used / (1024.0 ** 3)
+    memory_total = memory.total / (1024.0 ** 3)
+    return {"percent": f"{memory_percent}%", "used": f"{memory_used:.3f}GB", "total": f"{memory_total:.3f}GB"}
+
+
 
 def create_demo_keypose(process):
     with gr.Blocks() as demo:
@@ -34,6 +43,13 @@ def create_demo_sketch(process):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('T2I-Adapter (Sketch)')
+        
+        with gr.Row():
+            with gr.Column():
+                gr.Textbox(value="Hello Memory")
+            with gr.Column():
+                gr.JSON(get_system_memory, every=1)
+
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(source='upload', type="numpy")