Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def sentence_prediction(sentence):
|
|
47 |
test_data_loader = torch.utils.data.DataLoader(
|
48 |
test_dataset,
|
49 |
batch_size=config.VALID_BATCH_SIZE,
|
50 |
-
num_workers
|
51 |
)
|
52 |
|
53 |
device = config.device
|
|
|
47 |
test_data_loader = torch.utils.data.DataLoader(
|
48 |
test_dataset,
|
49 |
batch_size=config.VALID_BATCH_SIZE,
|
50 |
+
num_workers=-1
|
51 |
)
|
52 |
|
53 |
device = config.device
|