Spaces:
Sleeping
Sleeping
Update func.py
Browse files
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
|