Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import cohere | |
| import os | |
| import re | |
| import uuid | |
| from functools import partial | |
| from urllib.error import HTTPError | |
| cohere_api_key = os.getenv("COHERE_API_KEY") | |
| co = cohere.Client(cohere_api_key) | |
| history = [] | |
| chat = [] | |
| def trigger_example(example): | |
| chat, updated_history = generate_response(example) | |
| return chat, updated_history | |
| def generate_response(user_message, cid, history=None): | |
| if history is None: | |
| history = [] | |
| history.append(user_message) | |
| stream = co.chat_stream(message=user_message, conversation_id=cid, model='command-r', connectors=[], temperature=0.3) | |
| output = "" | |
| for idx, response in enumerate(stream): | |
| if response.event_type == "text-generation": | |
| output += response.text | |
| if idx == 0: | |
| history.append(" " + output) | |
| else: | |
| history[-1] = output | |
| chat = [ | |
| (history[i].strip(), history[i + 1].strip()) | |
| for i in range(0, len(history) - 1, 2) | |
| ] | |
| yield chat, history, cid | |
| return chat, history, cid | |
| def clear_chat(): | |
| return [], [], str(uuid.uuid4()) | |
| examples = [ | |
| "What are 8 good questions to get to know a stranger?", | |
| "Create a list of 10 unusual excuses people might use to get out of a work meeting", | |
| "Write a python code to reverse a string", | |
| "Explain the relativity theory in French", | |
| "Como sair de um helicóptero que caiu na água?", | |
| "Formally introduce the transformer architecture with notation.", | |
| "¿Cómo le explicarías el aprendizaje automático a un extraterrestre?", | |
| "Summarize recent news about the North American tech job market", | |
| "My coworker brought some delicious treats from their recent trip to the office to share. Would it be immoral if I took most of not all of these treats?", | |
| "Explain gravity to a chicken." | |
| ] | |
| title = """<h1 align="center">Cohere for AI Command R</h1>""" | |
| custom_css = """ | |
| #logo-img { | |
| border: none !important; | |
| } | |
| #chat-message { | |
| font-size: 14px; | |
| min-height: 300px; | |
| } | |
| """ | |
| with gr.Blocks(analytics_enabled=False, css=custom_css) as demo: | |
| #gr.HTML(title) | |
| cid = gr.State(str(uuid.uuid4())) | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| gr.Image("logo2.png", elem_id="logo-img", show_label=False, show_share_button=False, show_download_button=False) | |
| with gr.Column(scale=3): | |
| gr.Markdown("""C4AI Command R is a research release of a 35 billion parameter highly performant generative model. C4AI Command R is a large language model with open weights optimized for a variety of use cases including reasoning, summarization, and question answering. Command R has the capability for multilingual generation evaluated in 10 languages and highly performant RAG and tool use capabilities. | |
| <br/><br/> | |
| **Model**: [c4ai-command-r-v01](https://huggingface.co/CohereForAI/c4ai-command-r-v01) | |
| <br/> | |
| **Developed by**: [Cohere](https://cohere.com/) and [Cohere for AI](https://cohere.com/research) | |
| <br/> | |
| **License**: [CC-BY-NC](https://cohere.com/c4ai-cc-by-nc-license), requires also adhering to [C4AI's Acceptable Use Policy](https://docs.cohere.com/docs/c4ai-acceptable-use-policy) | |
| """ | |
| ) | |
| with gr.Column(): | |
| with gr.Row(): | |
| chatbot = gr.Chatbot(show_label=False) | |
| with gr.Row(): | |
| user_message = gr.Textbox(lines=1, placeholder="Ask anything ...", label="Input", show_label=False) | |
| with gr.Row(): | |
| submit_button = gr.Button("Submit") | |
| clear_button = gr.Button("Clear chat") | |
| history = gr.State([]) | |
| user_message.submit(fn=generate_response, inputs=[user_message, cid, history], outputs=[chatbot, history, cid], concurrency_limit=32) | |
| submit_button.click(fn=generate_response, inputs=[user_message, cid, history], outputs=[chatbot, history, cid], concurrency_limit=32) | |
| clear_button.click(fn=clear_chat, inputs=None, outputs=[chatbot, history, cid], concurrency_limit=32) | |
| user_message.submit(lambda x: gr.update(value=""), None, [user_message], queue=False) | |
| submit_button.click(lambda x: gr.update(value=""), None, [user_message], queue=False) | |
| clear_button.click(lambda x: gr.update(value=""), None, [user_message], queue=False) | |
| with gr.Row(): | |
| gr.Examples( | |
| examples=examples, | |
| inputs=[user_message], | |
| cache_examples=False, | |
| fn=trigger_example, | |
| outputs=[chatbot], | |
| ) | |
| if __name__ == "__main__": | |
| # demo.launch(debug=True) | |
| demo.queue(api_open=False).launch() |