memory debug

app.py

@@ -32,6 +32,18 @@ pipe.enable_model_cpu_offload()
 canny = CannyDetector()
 
 
+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",
+    }
+
+
 @spaces.GPU(duration=160)
 def inpaint(
     image,
@@ -60,6 +72,11 @@ def inpaint(
 
 with gr.Blocks() as demo:
     # gr.LoginButton()
+    with gr.Row():
+        with gr.Column():
+            gr.Textbox(value="Hello Memory")
+        with gr.Column():
+            gr.JSON(get_system_memory, every=1)
     gr.Interface(
         fn=inpaint,
         inputs=[
@@ -76,7 +93,7 @@ with gr.Blocks() as demo:
         description="Upload an image and a mask, then provide a prompt to generate an inpainted image.",
     )
 
-demo.launch()
+demo.launch(debug=True)
 
 # import gradio as gr
 # import numpy as np