paragon-analytics commited on
Commit
b20fb1e
·
1 Parent(s): f6468ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -117,7 +117,7 @@ def process_final_text(text):
117
  inp_text = "paraphrase: " + X_test + " </s>"
118
 
119
  encoding = para_tokenizer.encode_plus(inp_text,pad_to_max_length=True, return_tensors="pt")
120
- input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
121
 
122
  outputs = para_model.generate(
123
  input_ids=input_ids, attention_mask=attention_masks,
 
117
  inp_text = "paraphrase: " + X_test + " </s>"
118
 
119
  encoding = para_tokenizer.encode_plus(inp_text,pad_to_max_length=True, return_tensors="pt")
120
+ input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
121
 
122
  outputs = para_model.generate(
123
  input_ids=input_ids, attention_mask=attention_masks,