Emil25 commited on
Commit
293fc6f
·
verified ·
1 Parent(s): 22eee74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def gen_review(input_text):
35
  input_ids = tokenizer.encode(input_text, return_tensors='pt')
36
  output = model.generate(
37
  input_ids,
38
- max_length=300,
39
  num_return_sequences=1,
40
  no_repeat_ngram_size=2,
41
  do_sample=True,
 
35
  input_ids = tokenizer.encode(input_text, return_tensors='pt')
36
  output = model.generate(
37
  input_ids,
38
+ max_length=200,
39
  num_return_sequences=1,
40
  no_repeat_ngram_size=2,
41
  do_sample=True,