Update app.py

app.py

@@ -127,12 +127,8 @@ def prompt_model(model_name, system_prompt, user_prompt, sql_schema):
 
 def load_model(model_name):
     model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
-    #tokenizer = AutoTokenizer.from_pretrained(model_name)
-    tokenizer = model.tokenizer
     
-    model.generation_config.pad_token_id = tokenizer.pad_token_id
-   
-    return model, tokenizer
+    return model, model.tokenizer
     
 demo = gr.Interface(fn=process, 
                     inputs=[gr.Radio([ACTION_1, ACTION_2, ACTION_3], label = "Action", value = ACTION_3),