ajeetkumar01 commited on
Commit
0e7d5d7
·
verified ·
1 Parent(s): f9a03ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9
 
10
 
11
- def generate_text(input_text, max_length=16, num_beams=5, do_sample=False, no_repeat_ngram_size=2):
12
  """
13
  Generate text based on the given input text.
14
  Parameters:
 
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9
 
10
 
11
+ def generate_text(input_text, max_length=32, num_beams=5, do_sample=False, no_repeat_ngram_size=2):
12
  """
13
  Generate text based on the given input text.
14
  Parameters: