Update chain_app.py

chain_app.py

@@ -194,6 +194,10 @@ async def chat_profile():
             name="zephyr-7B",
             markdown_description="Open Source model Zephyr with 7B parameters",
         ),
+        cl.ChatProfile(
+            name='mistral-nemo-12B',
+            markdown_description='mistral open source LLM with 12B parameters'
+        ),
         cl.ChatProfile(
             name='mixtral-8x7B',
             markdown_description = 'mistral open source LLM with 7B parameters'
@@ -1050,6 +1054,29 @@ async def on_chat_start():
         await cl.Message(
             content="Im Mistral. the small version of Mistral Family. i was configured by Artin Daneshvar and Sadra Noadoust, 2 iranian students to help you, how can i assist you today ? "
         ).send()
+
+    if chat_profile == 'mistral-nemo-12B':
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Mistral-Model",
+                    label="Mistral - Model",
+                    values=["mistral-nemo-12B"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send()
+        await cl.Message(
+            content="Im Mistral nemo 12B .i was configured by Artin Daneshvar and Sadra Noadoust, 2 iranian students to help you, how can i assist you today ? "
+        ).send()
     # if chat_profile == 'Toka-353M':
     #     await cl.ChatSettings(
     #         [
@@ -1572,6 +1599,18 @@ async def main(message: cl.Message):
 
         await cl.Message(content=complete_content).send() 
 
+    elif chat_profile == 'mistral-nemo-12B':
+        client = Client("0x7o/Mistral-Nemo-Instruct", hf_token=hf_token)
+        result = client.predict(
+    		query=message.content,
+    		system="You are a helpful AI chatbot made by two iranian boys named Artin Daneshvar and Sadra Noadoust",
+    		api_name="/model_chat"
+        )
+        await cl.Message(
+            content=result[1][0][1]
+
+        ).send()
+
     # elif chat_profile == 'Toka-353M':
     #     output = query({
     #     "inputs": message.content,