Update app.py

app.py

@@ -13,7 +13,7 @@ import cohere
 import numpy as np
 import pandas as pd
 import gradio as gr
-
+import os
 from sklearn.metrics.pairwise import cosine_similarity
 from annoy import AnnoyIndex
 import warnings
@@ -26,7 +26,8 @@ data_df = pd.read_csv('functions_data.csv')
 data_df['docstring'].fillna('not specified', inplace=True)
 
 # Paste your API key here. Remember to not share publicly
-api_key = API_KEY
+key = os.environ.get('API_KEY')
+api_key = key
 
 # Create and retrieve a Cohere API key from dashboard.cohere.ai/welcome/register
 co = cohere.Client(api_key)