Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import json
 import time
 ###import mymodule as mm
-import util as util
+import psutil
 
 
 #函数部分
@@ -57,6 +57,17 @@ def demo1(state, value):
     else:
         mytext.value = "谦虚使人进步"
 
+def mem_info():
+    # 获取内存信息
+    mem_info = psutil.virtual_memory()
+
+    # 剩余内存，单位为字节
+    remaining_memory = mem_info.available
+
+    # 将字节转换为更易读的单位（例如 GB）
+    remaining_memory_gb = remaining_memory / (1024 ** 3)
+    return remaining_memory_gb
+    
 #会话变量
 ss.session.method = 0
 ss.session.prompt1 = "请将文言文拷贝到这里"
@@ -78,6 +89,11 @@ ss.button("开始翻译", onclick = predict)
 ss.space()
 
 ss.write("---")
-remaining_memory_gb = util.mem_info()
+
+###剩余内存测试
+remaining_memory_gb = mem_info()
 ss.write(f"剩余内存: {remaining_memory_gb:.2f} GB")
 
+
+
+