当前位置:   article > 正文

大模型对话页面设计gradio+LLM/API_gradio 大模型

gradio 大模型

背景:如今大模型火爆,接口众多,想要调用接口自己设计个界面进行测试,或者交给垂类专业人员校验大模型正确性,设计个页面比较好,当然gradio是最方便的,也是首选。要求:具有上下文信息,能够流式返回(增强体验效果),具有prompt功能。

1,实现上下文功能

需要维护一个history列表,严格说是两个,一个是模型需要的数据结构,一个是gradio需要的上下文。参考

  1. import gradio as gr
  2. import random
  3. import time
  4. with gr.Blocks() as demo:
  5. chatbot = gr.Chatbot()
  6. msg = gr.Textbox()
  7. clear = gr.ClearButton([msg, chatbot])
  8. def respond(message, chat_history):
  9. bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
  10. chat_history.append((message, bot_message))
  11. time.sleep(2)
  12. return "", chat_history
  13. msg.submit(respond, [msg, chatbot], [msg, chatbot])
  14. demo.la
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/人工智能uu/article/detail/881480
推荐阅读
相关标签
  

闽ICP备14008679号