Update app.py

app.py

@@ -2,10 +2,14 @@ import cohere
 import gradio as gr
 import os, json
 import random
-from transformers import pipeline
+
 import torch
 
 co = cohere.Client(os.environ.get('COHERE_API_KEY'))
+# initialize a conversation session id
+cohere_chat_res_start = co.chat("Hi")
+conv_session_id = cohere_chat_res_start.session_id
+
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")