updated prompt; reduce `max_token`; just return the whole text

app.py

@@ -11,7 +11,7 @@ co = cohere.Client(cohere_api_key)
 
 def generate_data(data_src):
 
-    command_prompt= f'''This is a sample dataset in csv below, and I want you to help me generate more data with different variations (at least 100 examples) 
+    command_prompt= f'''This is a sample dataset in csv below, and I want you to help me generate more data in csv format with different variations (at least 100 examples).
 
     ```csv
     {data_src}
@@ -20,16 +20,13 @@ def generate_data(data_src):
     response = co.generate(
                   model='command',
                   prompt=command_prompt,
-                  max_tokens=2606,
+                  max_tokens=260,
                   temperature=0.9,
                   k=0,
                   stop_sequences=[],
                   return_likelihoods='NONE')
     
-    txt_response = response.generations[0].text
-    txt_split_1 = txt_response.split("```csv")
-    txt_split_2 = txt_split_1[1].split("```")
-    data_text = txt_split_2[0]
+    data_text = response.generations[0].text
 
     return data_text