Peter commited on
Commit
3ca941f
1 Parent(s): ecba037

update defaults

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -125,7 +125,7 @@ def load_examples(examples_dir="examples"):
125
  for example in examples:
126
  with open(example, "r") as f:
127
  text = f.read()
128
- text_examples.append([text, "base", 4, 512, 0.7, 3.5, 3])
129
 
130
  return text_examples
131
 
@@ -149,7 +149,7 @@ if __name__ == "__main__":
149
  choices=["base", "large"], label="model size", default="base"
150
  ),
151
  gr.inputs.Slider(
152
- minimum=1, maximum=6, label="num_beams", default=4, step=1
153
  ),
154
  gr.inputs.Slider(
155
  minimum=512,
@@ -173,10 +173,10 @@ if __name__ == "__main__":
173
  ),
174
  ],
175
  outputs="html",
176
- examples_per_page=4,
177
  title=title,
178
  description=description,
179
  article="The model can be used with tag [pszemraj/led-large-book-summary](https://huggingface.co/pszemraj/led-large-book-summary). See the model card for details on usage & a notebook for a tutorial.",
180
  examples=load_examples(),
181
- cache_examples=False,
182
  ).launch()
 
125
  for example in examples:
126
  with open(example, "r") as f:
127
  text = f.read()
128
+ text_examples.append([text, "large", 1, 512, 0.7, 3.5, 3])
129
 
130
  return text_examples
131
 
 
149
  choices=["base", "large"], label="model size", default="base"
150
  ),
151
  gr.inputs.Slider(
152
+ minimum=1, maximum=4, label="num_beams", default=1, step=1
153
  ),
154
  gr.inputs.Slider(
155
  minimum=512,
 
173
  ),
174
  ],
175
  outputs="html",
176
+ examples_per_page=2,
177
  title=title,
178
  description=description,
179
  article="The model can be used with tag [pszemraj/led-large-book-summary](https://huggingface.co/pszemraj/led-large-book-summary). See the model card for details on usage & a notebook for a tutorial.",
180
  examples=load_examples(),
181
+ cache_examples=True,
182
  ).launch()