使用MongoDB构建AI：Story Tools Studio将生成式AI引入Myth Maker AI游戏

from transformers import AutoTokenizer, AutoModelForCausalLM
from story_tools_studio.utils.ai_utils import AIUtils
 
# 假设你已经有了一个预训练的GPT-4模型和tokenizer
tokenizer = AutoTokenizer.from_pretrained("tatsu-u/myth-maker-gpt4")
model = AutoModelForCausalLM.from_pretrained("tatsu-u/myth-maker-gpt4")
 
# 创建一个新的AIUtils实例
myth_maker = AIUtils(model, tokenizer)
 
# 使用Myth Maker生成文本
prompt = "In Ancient Greece, what was the most powerful deity?"
response = myth_maker.generate_text(prompt)
 
print(f"Prompt: {prompt}")
print(f"Response: {response}")

这段代码演示了如何使用Story Tools Studio的AI Utils来与Myth Maker AI进行交互。首先，我们加载了提前训练好的GPT-4模型和tokenizer。然后，我们创建了一个AIUtils实例，并使用它来生成响应。最后，我们打印出提示和生成的文本。这个例子简单明了地展示了如何将生成式AI集成到你的应用程序中。