Update app.py

app.py

@@ -18,8 +18,8 @@ cwi_tok = AutoTokenizer.from_pretrained('twigs/cwi-regressor')
 cwi_model = AutoModelForSequenceClassification.from_pretrained('twigs/cwi-regressor')
 simpl_tok = BartTokenizer.from_pretrained('twigs/bart-text2text-simplifier')
 simpl_model = BartForConditionalGeneration.from_pretrained('twigs/bart-text2text-simplifier')
-cwi_pipe = pipeline('text-classification', model=cwi_model, tokenizer=cwi_tok, function_to_apply='none', device=0)
-fill_pipe = pipeline('fill-mask', model=simpl_model, tokenizer=simpl_tok, top_k=1, device=0)
+cwi_pipe = pipeline('text-classification', model=cwi_model, tokenizer=cwi_tok, function_to_apply='none')
+fill_pipe = pipeline('fill-mask', model=simpl_model, tokenizer=simpl_tok, top_k=1)
 
 
 def id_replace_complex(s, threshold=0.4):
@@ -44,7 +44,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
+  out = simpl_tok([s], max_length=256, padding="max_length",  truncation=True, return_tensors='pt') if not tokenized else s
 
   generated_ids = model.generate(
       input_ids=out['input_ids'],