DiegoTheExplorar commited on
Commit
3d607e7
·
verified ·
1 Parent(s): 29f1bcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ n_layers = 2
12
  emb_dim = 256
13
  hid_dim = 512
14
  dropout = 0.5
15
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # Use GPU if available, otherwise use CPU
16
 
17
  # Load preprocessed data and model parameters
18
  (english_tokenizer, klingon_tokenizer, max_english_length,
 
12
  emb_dim = 256
13
  hid_dim = 512
14
  dropout = 0.5
15
+ device = torch.device('cpu') # Use GPU if available, otherwise use CPU
16
 
17
  # Load preprocessed data and model parameters
18
  (english_tokenizer, klingon_tokenizer, max_english_length,