kusumakar commited on
Commit
87e14b3
1 Parent(s): 82a0b71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def generate_paragraph(prompt):
27
  input_ids = tokenizer_1.encode(prompt, return_tensors="pt")
28
 
29
  # Generate the paragraph
30
- output = model_2.generate(input_ids, max_length=200, num_return_sequences=1, early_stopping=True)
31
 
32
  # Decode the generated output into text
33
  paragraph = tokenizer_1.decode(output[0], skip_special_tokens=True)
 
27
  input_ids = tokenizer_1.encode(prompt, return_tensors="pt")
28
 
29
  # Generate the paragraph
30
+ output = model_2.generate(input_ids, max_length=200, num_return_sequences=1, no_repeat_ngram_size=2, early_stopping=True)
31
 
32
  # Decode the generated output into text
33
  paragraph = tokenizer_1.decode(output[0], skip_special_tokens=True)