We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9031c5e commit f4c5380Copy full SHA for f4c5380
examples/as_app/talk_bot_AI/src/main.py
@@ -23,7 +23,7 @@ def ai_talk_bot_process_request(message: talk_bot.TalkBotMessage):
23
r = re.search(r"@ai\s(.*)", message.object_content["message"], re.IGNORECASE)
24
if r is None:
25
return
26
- model = pipeline("text2text-generation", model="MBZUAI/LaMini-Flan-T5-77M")
+ model = pipeline("text2text-generation", model=MODEL_NAME)
27
response_text = model(r.group(1), max_length=64, do_sample=True)[0]["generated_text"]
28
AI_BOT.send_message(response_text, message)
29
0 commit comments