pragnakalp commited on
Commit
66e7217
1 Parent(s): e55ec6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -41,8 +41,10 @@ def ocr_with_paddle(img):
41
  """
42
  Keras OCR
43
  """
 
44
  @spaces.GPU
45
  def ocr_with_keras(img):
 
46
  output_text = ''
47
  pipeline=keras_ocr.pipeline.Pipeline().to_device()
48
  images=[keras_ocr.tools.read(img)]
 
41
  """
42
  Keras OCR
43
  """
44
+ print("\n\n Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
45
  @spaces.GPU
46
  def ocr_with_keras(img):
47
+ print("\n\n inside Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
48
  output_text = ''
49
  pipeline=keras_ocr.pipeline.Pipeline().to_device()
50
  images=[keras_ocr.tools.read(img)]