Bils commited on
Commit
67ceb1e
·
verified ·
1 Parent(s): 7eb998e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -12,6 +12,8 @@ from pathlib import Path
12
  print(f"Is CUDA available: {torch.cuda.is_available()}")
13
  if torch.cuda.is_available():
14
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 
 
15
 
16
  # Load environment variables from .env file
17
  load_dotenv()
 
12
  print(f"Is CUDA available: {torch.cuda.is_available()}")
13
  if torch.cuda.is_available():
14
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
15
+ else:
16
+ print("Using CPU as fallback.")
17
 
18
  # Load environment variables from .env file
19
  load_dotenv()