fix

app.py

@@ -74,18 +74,19 @@ def update_gpu_status():
 
     except Exception as e:
         print(f"Error getting GPU stats: {e}")
-
-# def 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)
-#         used_memory = (gpu_memory[1] - gpu_memory[0]) / (1024 * 1024)
+        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)
+        used_memory = (gpu_memory[1] - gpu_memory[0]) / (1024 * 1024)
         
-#         gpu_status = f"GPU: {gpu_info}\nTotal Memory: {total_memory:.2f} MB\nUsed Memory: {used_memory:.2f} MB"
-#     else:
-#         gpu_status = "No GPU available"
-#     return gpu_status
+        gpu_status = f"GPU: {gpu_info}\nTotal Memory: {total_memory:.2f} MB\nUsed Memory: {used_memory:.2f} MB"
+    else:
+        gpu_status = "No GPU available"
+    return gpu_status
 
 def update_cpu_status():
     import datetime