Update app.py
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ class update_visibility:
|
|
| 41 |
return gr.UploadButton(visible=bool(1)) #make it visible
|
| 42 |
else:
|
| 43 |
return gr.UploadButton(visible=bool(0))
|
| 44 |
-
|
| 45 |
def train(model_name,
|
| 46 |
inject_prompt,
|
| 47 |
dataset_predefined,
|
|
|
|
| 41 |
return gr.UploadButton(visible=bool(1)) #make it visible
|
| 42 |
else:
|
| 43 |
return gr.UploadButton(visible=bool(0))
|
| 44 |
+
@spaces.GPU
|
| 45 |
def train(model_name,
|
| 46 |
inject_prompt,
|
| 47 |
dataset_predefined,
|