Hugging Face's logo

Spaces:
E-Hospital
/
oop-deploy
Paused

App Files Community
E-Hospital commited on
Commit
9110acb
•
1 Parent(s): d235373

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ import logging
4
 
5
  logger = logging.getLogger()
6
 
7
- device = torch.device("cuda" if torch.cuda.is_available() else "CPU")
8
 
9
  logger.info("Getting Device")
10
  logger.info(device)
 
4
 
5
  logger = logging.getLogger()
6
 
7
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
 
9
  logger.info("Getting Device")
10
  logger.info(device)