pragnakalp commited on
Commit
ed59084
·
verified ·
1 Parent(s): 82ee5d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,9 +21,10 @@ import spaces
21
  import pytesseract
22
  from PIL import Image
23
 
24
- os.environ['CUDA_VISIBLE_DEVICES'] = "0"
25
 
26
  gpus = tf.config.experimental.list_physical_devices('GPU')
 
27
  if gpus:
28
  try:
29
  for gpu in gpus:
 
21
  import pytesseract
22
  from PIL import Image
23
 
24
+ os.environ['CUDA_VISIBLE_DEVICES'] = "1"
25
 
26
  gpus = tf.config.experimental.list_physical_devices('GPU')
27
+ print("\n\n gpus :",gpus)
28
  if gpus:
29
  try:
30
  for gpu in gpus: