tykiww commited on
Commit
058f0e8
1 Parent(s): 010d9e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,