Update app.py

app.py

@@ -15,8 +15,8 @@ system_prompt = "You are a text to SQL query translator. Given a question in Eng
 user_prompt = "What is the total trade value and average price for each trader and stock in the trade_history table?"
 schema = "CREATE TABLE trade_history (id INT, trader_id INT, stock VARCHAR(255), price DECIMAL(5,2), quantity INT, trade_time TIMESTAMP);"
 
-base_model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
-dataset = "b-mc2/sql-create-context" #"gretelai/synthetic_text_to_sql"
+base_model_id = "codellama/CodeLlama-7b-hf"
+dataset = "b-mc2/sql-create-context"
 
 def prompt_model(model_id, system_prompt, user_prompt, schema):
     pipe = pipeline("text-generation", 
@@ -35,10 +35,10 @@ def prompt_model(model_id, system_prompt, user_prompt, schema):
     return result
 
 def fine_tune_model(base_model_id, dataset):
-    #tokenizer = download_model(base_model_id)
-    download_dataset(dataset)
-    #fine_tuned_model_id = upload_model(base_model_id, tokenizer)
-    return "Done" #fine_tuned_model_id
+    tokenizer = download_model(base_model_id)
+    #download_dataset(dataset)
+    fine_tuned_model_id = upload_model(base_model_id, tokenizer)
+    return fine_tuned_model_id
         
 def download_model(base_model_id):
     tokenizer = AutoTokenizer.from_pretrained(base_model_id)