Spaces:
Sleeping
Sleeping
make sure integers
Browse files
app.py
CHANGED
@@ -65,7 +65,8 @@ def create_response_paraphrase(input_str, max_length,num_return_sequences):
|
|
65 |
|
66 |
encoding = paraphrase_tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
|
67 |
input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
|
68 |
-
|
|
|
69 |
outputs = paraphrase_model.generate(
|
70 |
input_ids=input_ids, attention_mask=attention_masks,
|
71 |
# max_length=256,
|
|
|
65 |
|
66 |
encoding = paraphrase_tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
|
67 |
input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
|
68 |
+
num_return_sequences = int(num_return_sequences) # Ensure this is an integer
|
69 |
+
max_length = int(max_length)
|
70 |
outputs = paraphrase_model.generate(
|
71 |
input_ids=input_ids, attention_mask=attention_masks,
|
72 |
# max_length=256,
|