Update README.md

README.md

@@ -33,7 +33,7 @@ model = AutoModelForCausalLM.from_pretrained('gretelai/text2table').to('cuda', d
 
 model.eval()
 
-INSTRUCTION_KEY = "### Instruction: Given the following prompt, generate a table. Each column should have random values."
+INSTRUCTION_KEY = "### Instruction: Given the following prompt, generate a table."
 RESPONSE_KEY = "### Response:"
 INTRO_BLURB = "Below is an instruction that describes a task. Write a response that appropriately completes the request."
 PROMPT_FOR_GENERATION_FORMAT = """{intro}
@@ -43,12 +43,12 @@ PROMPT_FOR_GENERATION_FORMAT = """{intro}
 """.format(
     intro=INTRO_BLURB,
     instruction_key=INSTRUCTION_KEY,
-    prompt_to_generate_table="{PROMPT}",
+    prompt_to_generate_table="{prompt_to_generate_table}",
     response_key=RESPONSE_KEY,
 )
 
 PROMPT = "Create a dataset with four columns: patient, sex, agegrp, bp_before and bp_after. The patient column is a numerical identifier, sex is the gender of the patient, agegrp is the age group of the patient, bp_before is the blood pressure (in mmHg) before a certain treatment, and bp_after is the blood pressure (in mmHg) after a certain treatment."
-inputs = PROMPT_FOR_GENERATION_FORMAT.format(instruction=instruction)
+inputs = PROMPT_FOR_GENERATION_FORMAT.format(prompt_to_generate_table=PROMPT)
 tokenizer.pad_token = tokenizer.eos_token
 input = tokenizer(inputs, return_tensors="pt").to('cuda')
 input_ids = input['input_ids']