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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -66,14 +66,15 @@ def predict(
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
  """
78
  proc_submission - a helper function for the gradio module to process submissions
79
 
@@ -356,7 +357,7 @@ if __name__ == "__main__":
356
  fn=proc_submission,
357
  inputs=[
358
  input_text,
359
- model_name,
360
  num_beams,
361
  token_batch_length,
362
  length_penalty,
 
66
 
67
  def proc_submission(
68
  input_text: str,
 
69
  num_beams: int =2,
70
  token_batch_length: int =1024,
71
  length_penalty: float = 0.7,
72
  repetition_penalty: float =3.5,
73
  no_repeat_ngram_size: int =3,
74
+ max_input_length: int = 2560):
75
+ model_name="pszemraj/led-large-book-summary"
76
+
77
+
78
  """
79
  proc_submission - a helper function for the gradio module to process submissions
80
 
 
357
  fn=proc_submission,
358
  inputs=[
359
  input_text,
360
+ #model_name,
361
  num_beams,
362
  token_batch_length,
363
  length_penalty,