Update chain_app.py

chain_app.py

@@ -10,11 +10,10 @@ from huggingface_hub import InferenceClient
 
 
 hf_token = os.environ.get("HF_TOKEN")
-print(hf_token)
 openai_api_key = os.environ.get('OPENAI_API_KEY')
 groq_api_key = os.environ.get('GROQ_API_KEY')
 cohere_api_key = os.environ.get('COHERE_API_KEY')
-print(cohere_api_key)
+
 
 hf_text_client = Client("Artin2009/text-generation", hf_token=hf_token)
 # hf_image_client = Client('Artin2009/image-generation')
@@ -22,7 +21,7 @@ openai_client = OpenAI(api_key=openai_api_key)
 groq_client = Groq(api_key=groq_api_key)
 
 co = cohere.Client(
-  api_key=cohere_api_key, # This is your trial API key
+  api_key=cohere_api_key,
 ) 
 
 # API_URL = "https://api-inference.huggingface.co/models/PartAI/TookaBERT-Large"
@@ -629,7 +628,7 @@ async def main(message: cl.Message):
     elif chat_profile == 'Llama-3.1-405B':        
         client = InferenceClient(
             "meta-llama/Meta-Llama-3.1-405B-Instruct",
-            token="hf_token",
+            token=hf_token,
         )
         
         for message in client.chat_completion(