Update app.py

app.py

@@ -21,12 +21,9 @@ action_1 = "Prompt base model"
 action_2 = "Prompt fine-tuned model"
 action_3 = "Fine-tune base model"
 
-schema = "CREATE TABLE trade_history (id INT, trader_id INT, stock VARCHAR(255), price DECIMAL(5,2), quantity INT, trade_time TIMESTAMP);"
-system_prompt = """You are a text to SQL query translator.
-                   Given a question in English, generate a SQL query based on the provided SCHEMA.
-                   Do not generate any additional text.
-                   SCHEMA: """
+system_prompt = "You are a text to SQL query translator. Given a question in English, generate a SQL query based on the provided SCHEMA. Do not generate any additional text. 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 = "meta-llama/Meta-Llama-3-8B-Instruct"
 fine_tuned_model_id = "bstraehle/Meta-Llama-3-8B-Instruct"