Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "defog/sqlcoder-7b-2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model_name = "premai-io/prem-1B-SQL"
+tokenizer = AutoTokenizer.from_pretrained("premai-io/prem-1B-SQL")
+model = AutoModelForCausalLM.from_pretrained("premai-io/prem-1B-SQL")
 
 def generate_sql(natural_language_query):
     # Define your SQL generation logic here