Update app.py

app.py

@@ -34,6 +34,13 @@ async def generate_response(request_body: RequestBody):
             json_data = {'messages': [{'role': 'user','content': input_text}],'model': 'llama-3.1-70b-versatile',}
             response = requests.post('https://api.groq.com/openai/v1/chat/completions', headers=headers, json=json_data)
             all_chunk = response.json()["choices"][0]["message"]["content"]
+
+        if model == "cohere":
+            key_cohere = os.environ['key_cohere']
+            headers = {'accept': 'application/json','content-type': 'application/json','Authorization': f'Bearer {key_cohere}',}
+            data = {"model":"command-r-plus","messages":[{ "role": "user", "content": input_text}]}
+            response = requests.post('https://api.cohere.com/v2/chat', headers=headers, json=data)
+            all_chunk = response.json()['message']['content'][0]['text']
             
     if key_body != key_true:
         all_chunk = "How's the hack going?"