Hugging Face's logo

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

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -23,7 +23,6 @@ from PIL import Image
23
  import torch
24
 
25
  zero = torch.Tensor([0]).cuda()
26
- tf.config.experimental.set_memory_growth(zero, True)
27
 
28
  os.environ['CUDA_VISIBLE_DEVICES'] = zero
29
  print(zero.device)
 
23
  import torch
24
 
25
  zero = torch.Tensor([0]).cuda()
 
26
 
27
  os.environ['CUDA_VISIBLE_DEVICES'] = zero
28
  print(zero.device)