Hugging Face's logo

Spaces:
pragnakalp
/
OCR-image-to-text-ZeroGPU
Runtime error

App Files Community
pragnakalp commited on
Commit
be5cd6a
·
verified ·
1 Parent(s): baaf8bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -24,7 +24,6 @@ import torch
24
 
25
  zero = torch.Tensor([0]).cuda()
26
 
27
- os.environ['CUDA_VISIBLE_DEVICES'] = zero
28
  print(zero.device)
29
  # gpus = tf.config.experimental.list_physical_devices('GPU')
30
  # print("\n\n gpus :",gpus)
 
24
 
25
  zero = torch.Tensor([0]).cuda()
26
 
 
27
  print(zero.device)
28
  # gpus = tf.config.experimental.list_physical_devices('GPU')
29
  # print("\n\n gpus :",gpus)