Fix: total_mem -> total_memory for GPU info display

app.py

@@ -51,7 +51,7 @@ def get_gpu_info():
 
         if torch.cuda.is_available():
             props = torch.cuda.get_device_properties(0)
-            mem_total = props.total_mem / (1024**3)
+            mem_total = props.total_memory / (1024**3)
             mem_used = torch.cuda.memory_allocated(0) / (1024**3)
             mem_reserved = torch.cuda.memory_reserved(0) / (1024**3)
             return (