slliac commited on
Commit
89d0588
·
verified ·
1 Parent(s): beaeb40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def text2story(text):
52
  generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
53
  # Input should be of format "<BOS> <Genre token> Optional starter text"
54
  input_prompt = f"<BOS> <adventure> {text}"
55
- story = generator(input_prompt, min_length=50, max_length=100, do_sample=True,
56
  repetition_penalty=1.5, temperature=1.2,
57
  top_p=0.95, top_k=50)
58
  return story[0]['generated_text'].strip('<BOS> <adventure>')
 
52
  generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
53
  # Input should be of format "<BOS> <Genre token> Optional starter text"
54
  input_prompt = f"<BOS> <adventure> {text}"
55
+ story = generator(input_prompt, max_length=100, do_sample=True,
56
  repetition_penalty=1.5, temperature=1.2,
57
  top_p=0.95, top_k=50)
58
  return story[0]['generated_text'].strip('<BOS> <adventure>')