Omnibus commited on
Commit
8714c79
1 Parent(s): d3d3d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -66,12 +66,12 @@ def predict(
66
 
67
  def proc_submission(
68
  input_text: str,
69
- model_name: str,
70
- num_beams: int,
71
- token_batch_length: int,
72
- length_penalty: float,
73
- repetition_penalty: float,
74
- no_repeat_ngram_size: int,
75
  max_input_length: int = 2560,
76
  ):
77
  """
 
66
 
67
  def proc_submission(
68
  input_text: str,
69
+ model_name: str ="pszemraj/led-large-book-summary",
70
+ num_beams: int =2,
71
+ token_batch_length: int =1024,
72
+ length_penalty: float = 0.7,
73
+ repetition_penalty: float =3.5,
74
+ no_repeat_ngram_size: int =3,
75
  max_input_length: int = 2560,
76
  ):
77
  """