Update app.py

app.py

@@ -47,7 +47,7 @@ def id_replace_complex(s, threshold=0.4):
 def generate_candidate_text(s, model, tokenizer, tokenized=False):
 
   out = simpl_tok([s], max_length=256, padding="max_length",  truncation=True,
-                  return_tensors='pt').to('cuda') if not tokenized else s
+                  return_tensors='pt') if not tokenized else s
 
   generated_ids = model.generate(
       input_ids=out['input_ids'],