Spaces:
Running
Running
DiegoTheExplorar
commited on
Update app.py
Browse files
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('
|
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,
|