Update app.py

app.py

@@ -1,4 +1,4 @@
-import json; import gradio as gr; import requests as req
+import os; import json; import gradio as gr; import requests as req
 from strings import dfs_code, function_code, real_docstring, tree_code, insert_code, display_code, article_string, descr_string
 
 """
@@ -23,7 +23,10 @@ code_nl = "function for db connection"
 CT5_URL = "https://api-inference.huggingface.co/models/stmnk/codet5-small-code-summarization-python"
 CT5_METHOD = 'POST'
 API_URL = CT5_URL
-headers = {"Authorization": "Bearer api_UhCKXKyqxJOpOcbvrZurQFqmVNZRTtxVfl"}
+API_KEY = os.environ.get("API_KEY")
+
+# headers = {"Authorization": "Bearer api_UhCKXKyqxJOpOcbvrZurQFqmVNZRTtxVfl"}
+headers = {"Authorization": f"Bearer {API_KEY}"}
 
 def query(payload):
 	response = req.post(API_URL, headers=headers, json=payload)