Skip to content

Commit f4c5380

Browse files
committed
fixed typo in Transformers AI example
Signed-off-by: Alexander Piskun <bigcat88@icloud.com>
1 parent 9031c5e commit f4c5380

File tree

1 file changed

+1
-1
lines changed
  • examples/as_app/talk_bot_AI/src

1 file changed

+1
-1
lines changed

examples/as_app/talk_bot_AI/src/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def ai_talk_bot_process_request(message: talk_bot.TalkBotMessage):
2323
r = re.search(r"@ai\s(.*)", message.object_content["message"], re.IGNORECASE)
2424
if r is None:
2525
return
26-
model = pipeline("text2text-generation", model="MBZUAI/LaMini-Flan-T5-77M")
26+
model = pipeline("text2text-generation", model=MODEL_NAME)
2727
response_text = model(r.group(1), max_length=64, do_sample=True)[0]["generated_text"]
2828
AI_BOT.send_message(response_text, message)
2929

0 commit comments

Comments
 (0)