Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("
|
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("
|
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",
|