Update app.py

app.py

@@ -7,6 +7,7 @@ from PIL import Image
 import timm
 import io
 import sys
+import psutil # 記得 import 這個，才能調用後台硬體資料
 
 # --- 1. 初始化模型 ---
 model_id = "briaai/RMBG-2.0"
@@ -56,7 +57,15 @@ def process_image(input_image):
     image.putalpha(mask)
     return image
 
-# --- 3. 介面 ---
+# --- 3.取得系統狀態的函數 ---
+def get_system_stats():
+    # 取得 CPU 使用率
+    cpu = psutil.cpu_percent(interval=None)
+    # 取得 RAM 使用率
+    ram = psutil.virtual_memory().percent
+    return {"cpu": cpu, "ram": ram}
+
+# --- 4. 介面 ---
 with gr.Blocks(title="RMBG-2.0 AI") as app:
     gr.Markdown("## ✂️ 去背服務測試")