Update app.py

app.py

@@ -173,8 +173,8 @@ def predict(text, tokenizer=tokenizer):
   with torch.no_grad():
 
     result = model(
-      tokenized_example_text['input_ids'].to(device),
-      attention_mask = tokenized_example_text['attention_masks'].to(device),
+      tokenized_example_text['input_ids'].to('cpu'),
+      attention_mask = tokenized_example_text['attention_masks'].to('cpu'),
       return_dict=True
   ).logits