captchaboy commited on
Commit
cd87987
1 Parent(s): 87706f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -10,9 +10,6 @@ import torch, pickle, strhub
10
  from PIL import Image
11
  print(f"Is CUDA available: {torch.cuda.is_available()}")
12
 
13
- # True
14
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
15
-
16
 
17
  # from strhub.data.module import SceneTextDataModule
18
  # from strhub.models.utils import load_from_checkpoint, parse_model_args
 
10
  from PIL import Image
11
  print(f"Is CUDA available: {torch.cuda.is_available()}")
12
 
 
 
 
13
 
14
  # from strhub.data.module import SceneTextDataModule
15
  # from strhub.models.utils import load_from_checkpoint, parse_model_args