Spaces:
Sleeping
Sleeping
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,
|