Update app.py

app.py

@@ -7,12 +7,12 @@ from langchain.prompts import PromptTemplate
 import gradio as gr
 
 # Set up the ChatGroq API
-api_key = os.environ.get("GROQ_API_KEY")
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 if not api_key:
     raise ValueError("Please set the GROQ_API_KEY environment variable")
 
 # Initialize the ChatGroq LLM
-llm = ChatGroq(temperature=0, model_name='llama-3.1-8b-instant', groq_api_key=api_key)
+llm = ChatGroq(temperature=0, model_name='llama-3.1-8b-instant', groq_api_key=GROQ_API_KEY)
 
 # Define a prompt template for entity extraction
 entity_extraction_prompt = PromptTemplate(