LinkLinkWu commited on
Commit
c212840
·
verified ·
1 Parent(s): b854469

Update func.py

Browse files
Files changed (1) hide show
  1. func.py +1 -1
func.py CHANGED
@@ -39,7 +39,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
39
  _MODEL_NAME = "aspis/gpt2-genre-story-generation"
40
  _PROMPT = (
41
  "Write a funny and warm children's story (50-100 words) for ages 3-10, "
42
- "fully based on this scene: {caption}\nStory:"
43
  )
44
 
45
  _tokenizer, _model = None, None
 
39
  _MODEL_NAME = "aspis/gpt2-genre-story-generation"
40
  _PROMPT = (
41
  "Write a funny and warm children's story (50-100 words) for ages 3-10, "
42
+ "fully and strictly based on this scene: {caption}\nStory:"
43
  )
44
 
45
  _tokenizer, _model = None, None