Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def classify_GPU(input_data):
|
|
| 31 |
|
| 32 |
SEQ_LEN = 1024
|
| 33 |
PAD_IDX = 14627
|
| 34 |
-
DEVICE = '
|
| 35 |
|
| 36 |
# instantiate the model
|
| 37 |
|
|
|
|
| 31 |
|
| 32 |
SEQ_LEN = 1024
|
| 33 |
PAD_IDX = 14627
|
| 34 |
+
DEVICE = 'cpu' # 'cuda'
|
| 35 |
|
| 36 |
# instantiate the model
|
| 37 |
|