Update app.py

app.py

@@ -18,7 +18,7 @@ import datetime
 import psutil
 import subprocess
 from gpustat import GPUStatCollection
-import platform
+import cpuinfo
 
 try:
     import spaces
@@ -74,18 +74,17 @@ def update_gpu_status():
             return gpu_status
 
     except Exception as e:
-        print(f"Error getting GPU stats: {e}")
         return torch_update_gpu_status()
 
 def torch_update_gpu_status():
     if torch.cuda.is_available():
         gpu_info = torch.cuda.get_device_name(0)
         gpu_memory = torch.cuda.mem_get_info(0)
-        total_memory = gpu_memory[1] / (1024 * 1024)
-        free_memory=gpu_memory[0] /(1024 *1024)
-        used_memory = (gpu_memory[1] - gpu_memory[0]) / (1024 * 1024)
+        total_memory = gpu_memory[1] / (1024 * 1024 * 1024)
+        free_memory=gpu_memory[0] /(1024 *1024 * 1024)
+        used_memory = (gpu_memory[1] - gpu_memory[0]) / (1024 * 1024 * 1024)
         
-        gpu_status = f"**GPU Name**: {gpu_info}\n**Free Memory**: {free_memory}MB\n**Total Memory**: {total_memory:.2f} MB\n**Used Memory**: {used_memory:.2f} MB\n"
+        gpu_status = f"**GPU Name**: {gpu_info}\n**Free Memory**: {free_memory}GB\n**Total Memory**: {total_memory:.2f} GB\n**Used Memory**: {used_memory:.2f} GB\n"
     else:
         gpu_status = "No GPU available"
     return gpu_status
@@ -97,15 +96,15 @@ def update_cpu_status():
     cpu_percent = psutil.cpu_percent()
     cpu_freq = psutil.cpu_freq()
     cpu_count = psutil.cpu_count(logical=True)
-    cpu_name = platform.processor()
+    cpu_name = cpuinfo.get_cpu_info().get("brand_raw", "Unknown CPU")
     virtual_memory = psutil.virtual_memory()
 
     cpu_status = f"**{time_str} (UTC+0)**\n\n"
     cpu_status += f"**CPU Name**: {cpu_name}\n"
     cpu_status += f"**CPU Usage**: {cpu_percent}%\n"
-    cpu_status += f"**CPU Frequency**: Current: {cpu_freq.current}MHz, Max: {cpu_freq.max}MHz, Min: {cpu_freq.min}MHz\n"
+    cpu_status += f"**CPU Frequency**: *Current*: {cpu_freq.current:.2f}MHz, *Max*: {cpu_freq.max:.2f}MHz, *Min*: {cpu_freq.min:.2f}MHz\n"
     cpu_status += f"**CPU Cores**: {cpu_count}\n"
-    cpu_status += f"**Virtual Memory**: Total: {virtual_memory.total}, Available: {virtual_memory.available}, Used: {virtual_memory.used}, Percentage: {virtual_memory.percent}%\n"
+    cpu_status += f"**Virtual Memory**: *Total*: {virtual_memory.total / (1024 * 1024 * 1024)}GB, *Available*: {virtual_memory.available / (1024 * 1024 * 1024)}GB, *Used*: {virtual_memory.used / (1024 * 1024 * 1024)}GB, *Percentage*: {virtual_memory.percent}%\n\n"
 
     return cpu_status