Rahatara commited on
Commit
c774f33
1 Parent(s): 039c073

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ def create_downloadable_zip(augmented_images):
40
  zip_buffer.seek(0)
41
  return zip_buffer
42
 
43
- st.title("Batch Image Augmentation for Dataset Creation")
44
 
45
  uploaded_files = st.file_uploader("Choose images (1-10)", accept_multiple_files=True, type=["jpg", "jpeg", "png"])
46
  augmentations_count = st.number_input("Number of augmented samples per image", min_value=1, max_value=10, value=3)
@@ -52,7 +52,7 @@ if uploaded_files:
52
  augmented_images = augment_image(image, augmentations_count)
53
  all_augmented_images.extend(augmented_images)
54
 
55
- if st.button("Download Augmented Dataset") and all_augmented_images:
56
  zip_buffer = create_downloadable_zip(all_augmented_images)
57
  st.download_button(
58
  label="Download ZIP",
 
40
  zip_buffer.seek(0)
41
  return zip_buffer
42
 
43
+ st.title("Ready-To-Use Synthetic Image Dataset Generation with Few-shots")
44
 
45
  uploaded_files = st.file_uploader("Choose images (1-10)", accept_multiple_files=True, type=["jpg", "jpeg", "png"])
46
  augmentations_count = st.number_input("Number of augmented samples per image", min_value=1, max_value=10, value=3)
 
52
  augmented_images = augment_image(image, augmentations_count)
53
  all_augmented_images.extend(augmented_images)
54
 
55
+ if st.button("Generate Synthetic Dataset") and all_augmented_images:
56
  zip_buffer = create_downloadable_zip(all_augmented_images)
57
  st.download_button(
58
  label="Download ZIP",