NimaKL commited on
Commit
b6cfb6c
β€’
1 Parent(s): 12c3915

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def preprocessing(input_text, tokenizer):
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
 
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