saiful9379 commited on
Commit
234e9f7
1 Parent(s): 4e70c1c

change device to cpu

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ from utils import utils_cls
7
  from model import BanglaTransformer
8
  from config import config as cfg
9
  torch.manual_seed(0)
10
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
11
- # device = torch.device('cpu')
12
  uobj = utils_cls(device=device)
13
 
14
  __MODULE__ = "Bangla Language Translation"
 
7
  from model import BanglaTransformer
8
  from config import config as cfg
9
  torch.manual_seed(0)
10
+ # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
11
+ device = torch.device('cpu')
12
  uobj = utils_cls(device=device)
13
 
14
  __MODULE__ = "Bangla Language Translation"