panda1835 commited on
Commit
dbc83fe
·
1 Parent(s): 779f50a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -13,9 +13,7 @@ from datetime import datetime
13
  import models
14
 
15
  print(f"Is CUDA available: {torch.cuda.is_available()}")
16
- # True
17
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
18
- # Tesla T4
19
 
20
  with open("index_to_species.json", "r") as file:
21
  index_to_species_data = file.read()
 
13
  import models
14
 
15
  print(f"Is CUDA available: {torch.cuda.is_available()}")
16
+ # print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 
 
17
 
18
  with open("index_to_species.json", "r") as file:
19
  index_to_species_data = file.read()