Update app.py

app.py

@@ -1,15 +1,18 @@
 import streamlit as st
 import pickle
 import cohere
+import os
 
 # Initialize Cohere Client
-co = cohere.Client('mBS7SXo0I577gd5g5lScX3rw1YKupJHQq866YvS6')  # Replace with your Cohere API key
+api_key = os.environ["API_KEY"]
+co = cohere.Client('api_key')  # Replace with your Cohere API key
 
 # Function to generate text using Cohere API
 # Function to generate text using Cohere API
 def generate_text(prompt, temperature):
     response = co.generate(
-        model='7f4b5dda-dcf2-40ca-9d4c-afdcaf435480-ft',  # Replace with your Cohere model ID
+        model_id = os.environ["MODEL_ID"]
+        model='model_id',  # Replace with your Cohere model ID
         prompt=prompt,
         max_tokens= 2000,
         temperature=temperature,