Shuu12121 commited on
Commit
68d0547
·
verified ·
1 Parent(s): 569f6b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,8 +72,8 @@ def generate_docstring(code: str) -> str:
72
  output_ids = model.generate(
73
  input_ids=inputs.input_ids,
74
  attention_mask=inputs.attention_mask,
75
- max_length=128,
76
- num_beams=4,
77
  early_stopping=True,
78
  eos_token_id=decoder_tokenizer.eos_token_id,
79
  pad_token_id=pad_token_id,
 
72
  output_ids = model.generate(
73
  input_ids=inputs.input_ids,
74
  attention_mask=inputs.attention_mask,
75
+ max_length=256,
76
+ num_beams=10,
77
  early_stopping=True,
78
  eos_token_id=decoder_tokenizer.eos_token_id,
79
  pad_token_id=pad_token_id,