zmbfeng commited on
Commit
87ff15f
·
1 Parent(s): e7b8dfa

make sure integers

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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,