milyiyo commited on
Commit
d8be3b1
1 Parent(s): 546da88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -82,7 +82,8 @@ def paraphrase(sentence: str, count: str):
82
  return {'result': []}
83
  sentence_input = sentence
84
  text = f"paraphrase: {sentence_input} </s>"
85
- encoding = tokenizer.encode_plus(text, padding=True, return_tensors="pt")
 
86
  # input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
87
 
88
  # outputs = model.generate(
 
82
  return {'result': []}
83
  sentence_input = sentence
84
  text = f"paraphrase: {sentence_input} </s>"
85
+ # encoding = tokenizer.encode_plus(text, padding=True, return_tensors="pt")
86
+ encoding = tokenizer(text, return_tensors="pt")
87
  # input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
88
 
89
  # outputs = model.generate(