Spaces:

bstraehle
/

sft

Running

bstraehle commited on Jul 8

Commit

15b743b

•

1 Parent(s): e566aba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ 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 = "gretelai/synthetic_text_to_sql"
 def prompt_model(model_id, system_prompt, user_prompt, schema):

 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 = "codellama/CodeLlama-7b-hf"
 dataset = "gretelai/synthetic_text_to_sql"
 def prompt_model(model_id, system_prompt, user_prompt, schema):