Changing mode to CPU execution

app.py

@@ -16,7 +16,7 @@ from simpletransformers.ner import NERModel, NERArgs
 HTML_WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
 
 
-@st.cache(allow_output_mutation=true)
+@st.cache(allow_output_mutation=True)
 def load_models():
     """Load custom built NER4OPT model."""
     custom_labels = [
@@ -52,12 +52,12 @@ def load_models():
     model_args.eval_batch_size = 1
     model_args.manual_seed = 123456789
     model_args.output_dir = "trained_transformer_model"
-    model_args.use_cuda = True
+    model_args.use_cuda = False
     model_args.use_multiprocessing = False
     model = NERModel("roberta",
                      "skadio/ner4opt-roberta",
                      labels=custom_labels,
-                     use_cuda=True,
+                     use_cuda=False,
                      args=model_args)
     spacy_model = spacy.load("en_core_web_sm")
     spacy_model.tokenizer = Tokenizer(spacy_model.vocab,