Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def preprocessing(input_text, tokenizer):
|
|
| 35 |
return_attention_mask = True,
|
| 36 |
return_tensors = 'pt'
|
| 37 |
)
|
| 38 |
-
|
| 39 |
|
| 40 |
def predict(new_sentence):
|
| 41 |
# We need Token IDs and Attention Mask for inference on the new sentence
|
|
|
|
| 35 |
return_attention_mask = True,
|
| 36 |
return_tensors = 'pt'
|
| 37 |
)
|
| 38 |
+
device = 'cpu'
|
| 39 |
|
| 40 |
def predict(new_sentence):
|
| 41 |
# We need Token IDs and Attention Mask for inference on the new sentence
|