Spaces:
Sleeping
Sleeping
Commit
·
b20fb1e
1
Parent(s):
f6468ff
Update app.py
Browse files
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"]
|
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,
|