Update app.py

app.py

@@ -17,8 +17,8 @@ SYSTEM_PROMPT = "You are a text to SQL query translator. Given a question in Eng
 USER_PROMPT = "How many new users joined from countries with stricter data privacy laws than the United States in the past month?"
 SQL_CONTEXT = "CREATE TABLE users (user_id INT, country VARCHAR(50), joined_date DATE); CREATE TABLE data_privacy_laws (country VARCHAR(50), privacy_level INT); INSERT INTO users (user_id, country, joined_date) VALUES (1, 'USA', '2023-02-15'), (2, 'Germany', '2023-02-27'); INSERT INTO data_privacy_laws (country, privacy_level) VALUES ('USA', 5), ('Germany', 8);"
 
-BASE_MODEL_NAME = "meta-llama/Meta-Llama-3.1-8B-Instruct"
-FT_MODEL_NAME = "Meta-Llama-3.1-8B-Instruct-text-to-sql"
+BASE_MODEL_NAME = "meta-llama/Meta-Llama-3.1-8B"
+FT_MODEL_NAME = "Meta-Llama-3.1-8B-text-to-sql"
 DATASET_NAME = "gretelai/synthetic_text_to_sql"
 
 def process(action, base_model_name, ft_model_name, dataset_name, system_prompt, user_prompt, sql_context):