Update app.py

app.py

@@ -35,7 +35,7 @@ def preprocessing(input_text, tokenizer):
         return_attention_mask = True,
         return_tensors = 'pt'
             )
-    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    device = 'cpu'
     
 def predict(new_sentence):
     # We need Token IDs and Attention Mask for inference on the new sentence