vram

app.py

@@ -12,7 +12,7 @@ def greet(n):
     if torch.cuda.is_available():
         total_memory_bytes = torch.cuda.get_device_properties(0).total_memory
         gb =  total_memory_bytes / (1024**3)
-        print(f"VRAM 0: {gb:.2f} GB"
+        print(f"VRAM 0: {gb:.2f} GB")
     else:
         print("CUDA is NOT available")
     return f"Hello {zero + n} Tensor"