Commit
·
f13c68c
1
Parent(s):
106ce34
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ from pathlib import Path
|
|
| 18 |
import spaces
|
| 19 |
import zipfile
|
| 20 |
|
| 21 |
-
MAX_IMAGES =
|
| 22 |
|
| 23 |
training_script_url = "https://raw.githubusercontent.com/huggingface/diffusers/main/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py"
|
| 24 |
subprocess.run(['wget', '-N', training_script_url])
|
|
@@ -85,6 +85,10 @@ num_images_settings = {
|
|
| 85 |
|
| 86 |
def load_captioning(uploaded_images, option):
|
| 87 |
updates = []
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
if len(uploaded_images) > MAX_IMAGES:
|
| 89 |
raise gr.Error(
|
| 90 |
f"Error: for now, only {MAX_IMAGES} or less images are allowed for training"
|
|
|
|
| 18 |
import spaces
|
| 19 |
import zipfile
|
| 20 |
|
| 21 |
+
MAX_IMAGES = 100
|
| 22 |
|
| 23 |
training_script_url = "https://raw.githubusercontent.com/huggingface/diffusers/main/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py"
|
| 24 |
subprocess.run(['wget', '-N', training_script_url])
|
|
|
|
| 85 |
|
| 86 |
def load_captioning(uploaded_images, option):
|
| 87 |
updates = []
|
| 88 |
+
if len(uploaded_images <= 1):
|
| 89 |
+
raise gr.Error(
|
| 90 |
+
"Error: please upload at least 2 images to train your model (the ideal number with default settings is between 4-30)"
|
| 91 |
+
)
|
| 92 |
if len(uploaded_images) > MAX_IMAGES:
|
| 93 |
raise gr.Error(
|
| 94 |
f"Error: for now, only {MAX_IMAGES} or less images are allowed for training"
|