Update app.py

app.py

@@ -42,12 +42,11 @@ class StopOnTokens(StoppingCriteria):
                 return True
         return False
 
-def bot(input_message: str, temperature=0.1, top_p=0.9, top_k=0, repetition_penalty=1.08):
-    def bot(input_message: str, temperature=0.1, top_p=0.9, top_k=0, repetition_penalty=1.08):
+def bot(input_message: str, db_info="", temperature=0.1, top_p=0.9, top_k=0, repetition_penalty=1.08):
     stop = StopOnTokens()
 
     # Format the user's input message
-    messages = f"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n\n### Instruction:\n\nConvert text to sql: {input_message}\n\n### Response:\n\n"
+    messages = f"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n\n### Instruction:\n\nConvert text to sql: {input_message} {db_info}\n\n### Response:\n\n"
 
     input_ids = tok(messages, return_tensors="pt").input_ids
     input_ids = input_ids.to(m.device)