Update README.md

README.md

@@ -37,7 +37,7 @@ The main purpose of GOAT-70B-STORYTELLING is to generate books, novels, moviescr
 
 Usage can be either self-hosted via `transformers` or used with Spaces
 
-```
+```python
 import torch
 
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -53,8 +53,10 @@ model = AutoModelForCausalLM.from_pretrained(
 
 Use it inside via GOAT-STORYTELLING-AGENT framework:
 
-```
-GOAT-STORYTELLING-AGENT code
+```python
+from goat_storytelling_agent.story_processor.prompt_manager import generate_story
+
+novel_scenes = generate_story('never too much coffee', form='novel')
 ```
 ## License