pls work

app.py

@@ -10,7 +10,7 @@ max_iters = 5000
 eval_interval = 500
 learning_rate = 3e-4
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-print(f"The code is running on {device} : GPU={torch.cuda.get_device_name(0)}")
+print(f"The code is running on {device}")
 eval_iters = 200
 n_embd = 384
 n_head = 6