Update app.py

app.py

@@ -25,10 +25,11 @@ if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
     # Detect if bf16 is enabled or not
     enable_bf16 = detect_bf16_support()
+    print(f'Device: {device}, GPU type: {gpu_type}')
+    print('BF16 enabled:', enable_bf16)
 else:
     device, power_device, enable_bf16 = "cpu", "CPU", False
-print(f'Device: {device}, GPU type: {gpu_type}')
-print('BF16 enabled:', enable_bf16)
+    print('Running on CPU')
 
 
 # The flag below controls whether to allow TF32 on matmul. This flag defaults to False in PyTorch 1.12 and later.