Update app.py

app.py

@@ -15,7 +15,6 @@ class SQLQueryGenerator:
         self.prompt_file = prompt_file
         self.metadata_file = metadata_file
     
-    @spaces.GPU
     def get_tokenizer_model(self, model_name):
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(
@@ -69,6 +68,7 @@ def main():
     model_name = "defog/sqlcoder-34b-alpha"
     sql_query_generator = SQLQueryGenerator(model_name)
 
+    @spaces.GPU
     def generate_sql(question):
         return sql_query_generator.run_inference(question)