Update app.py
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ def upload_and_prepare_dataset(files, dataset_name, trigger_word):
|
|
| 70 |
return f"✅ ready: {image_count} images in {dataset_name}", str(dataset_path), dataset_name
|
| 71 |
|
| 72 |
# request 10 mins gpu for training
|
| 73 |
-
@spaces.GPU(duration=
|
| 74 |
def train_lora(
|
| 75 |
dataset_path,
|
| 76 |
project_name,
|
|
|
|
| 70 |
return f"✅ ready: {image_count} images in {dataset_name}", str(dataset_path), dataset_name
|
| 71 |
|
| 72 |
# request 10 mins gpu for training
|
| 73 |
+
@spaces.GPU(duration=200)
|
| 74 |
def train_lora(
|
| 75 |
dataset_path,
|
| 76 |
project_name,
|