Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|
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>')
|