Alexander Watson commited on
Commit
097faaf
1 Parent(s): 476f41e
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -362,6 +362,9 @@ def main():
362
  st.session_state.response_format_prompt = response_format_prompt
363
 
364
  with st.expander("Download SDK Code", expanded=False):
 
 
 
365
  config_text = f"""
366
  #!pip install -Uqq git+https://github.com/gretelai/navigator-helpers.git
367
 
@@ -555,7 +558,7 @@ def main():
555
  # Write the augmented data to a file if it exists
556
  if augmented_data_jsonl:
557
  augmented_data_file_path = os.path.join(
558
- temp_dir, "augmented_data.jsonl"
559
  )
560
  with open(augmented_data_file_path, "w") as augmented_data_file:
561
  augmented_data_file.write(augmented_data_jsonl)
@@ -570,9 +573,9 @@ def main():
570
  # Download the ZIP file
571
  with open(zip_file_path, "rb") as zip_file:
572
  st.download_button(
573
- label="Download Synthetic Data and Logs",
574
  data=zip_file.read(),
575
- file_name="augmentation_results.zip",
576
  mime="application/zip",
577
  )
578
 
 
362
  st.session_state.response_format_prompt = response_format_prompt
363
 
364
  with st.expander("Download SDK Code", expanded=False):
365
+ st.markdown("### Ready to generate data at scale?")
366
+ st.write("Get started with your current configuration using the SDK code below:")
367
+
368
  config_text = f"""
369
  #!pip install -Uqq git+https://github.com/gretelai/navigator-helpers.git
370
 
 
558
  # Write the augmented data to a file if it exists
559
  if augmented_data_jsonl:
560
  augmented_data_file_path = os.path.join(
561
+ temp_dir, "synthetic_data.jsonl"
562
  )
563
  with open(augmented_data_file_path, "w") as augmented_data_file:
564
  augmented_data_file.write(augmented_data_jsonl)
 
573
  # Download the ZIP file
574
  with open(zip_file_path, "rb") as zip_file:
575
  st.download_button(
576
+ label="💾 Download Synthetic Data and Logs",
577
  data=zip_file.read(),
578
+ file_name="gretel_synthetic_data.zip",
579
  mime="application/zip",
580
  )
581