rohansampath commited on
Commit
e22c986
·
verified ·
1 Parent(s): 3601959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,8 @@ import pandas as pd
7
  import time
8
  import traceback
9
 
 
 
10
  # Read token and login
11
  hf_token = os.getenv("HF_READ_WRITE_TOKEN")
12
  if hf_token:
 
7
  import time
8
  import traceback
9
 
10
+ print ("IS CUDA AVAILABLE: ", torch.cuda.is_available())
11
+
12
  # Read token and login
13
  hf_token = os.getenv("HF_READ_WRITE_TOKEN")
14
  if hf_token: