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
|