Update app.py

app.py

@@ -145,7 +145,7 @@ def stats(text):
 def predict(text, tokenizer=tokenizer):
   
   model.eval()
-  model.to(device)
+  model.to('cpu')
   def prepare_data(text, tokenizer):
     
     input_ids = []