Update app.py

app.py

@@ -41,14 +41,14 @@ MODEL_NAME = "gpt-3.5-turbo"
 
 
 @st.cache
-def ask_model(doc, question):
+def ask_model(query):
     """This is the call to the OpenAI API. It creates a prompt from the document
     and question and returns the endpoint's response."""
 
     messages = BASE_CHAT_MESSAGES + [
         {
             "role": "user",
-            "content": f"Example SQL Query: {question.strip()}\n\n",
+            "content": f"Example SQL Query: {query.strip()}\n\n",
         }
     ]