Update app.py

app.py

@@ -64,9 +64,8 @@ class SQLQueryGenerator:
         )
         return generated_query
 
-
 @spaces.GPU  # Decorate this function to use GPU
-def generate_sql(question):
+def generate_sql(question, sql_query_generator):
     return sql_query_generator.run_inference(question)
 
 def main():
@@ -78,7 +77,7 @@ def main():
         question = gr.Textbox(label="Enter your question")
         submit = gr.Button("Generate SQL Query")
         output = gr.Textbox(label="🌬️🌁🌫️SqlCoder-2")
-        submit.click(fn=generate_sql, inputs=question, outputs=output)
+        submit.click(fn=generate_sql, inputs=[question, gr.State(sql_query_generator)], outputs=output)
 
     demo.launch()