Update app.py

app.py

@@ -144,7 +144,7 @@ def prompt_model(model_name, system_prompt, user_prompt, sql_context):
 
 def load_model(model_name):
     #model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
-    model = PeftModel.from_pretrained(model_name, device_map="auto")
+    model = PeftModel.from_pretrained(model_id=model_name, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.pad_token = tokenizer.eos_token