ibcplateformes Claude Opus 4.6 commited on
Commit
4b6c221
·
1 Parent(s): 259efa9

Import @spaces.GPU functions at top level for ZeroGPU detection

Browse files

ZeroGPU requires GPU-decorated functions to be importable at startup.
Moved training imports from lazy (inside handler) to top-level.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -85,6 +85,10 @@ else:
85
  )
86
 
87
 
 
 
 
 
88
  # ── Training Tab ─────────────────────────────────────────────────────────────
89
 
90
  def train_voice_model(audio_file, model_name, epochs, progress=gr.Progress()):
@@ -97,8 +101,6 @@ def train_voice_model(audio_file, model_name, epochs, progress=gr.Progress()):
97
 
98
  model_name = model_name.strip().replace(" ", "_")
99
 
100
- from pipeline.training import full_training_pipeline
101
-
102
  def progress_callback(value, desc):
103
  progress(value, desc=desc)
104
 
 
85
  )
86
 
87
 
88
+ # ── Import GPU-decorated functions at top level for ZeroGPU detection ───────
89
+ from pipeline.training import full_training_pipeline, extract_features, train_model
90
+
91
+
92
  # ── Training Tab ─────────────────────────────────────────────────────────────
93
 
94
  def train_voice_model(audio_file, model_name, epochs, progress=gr.Progress()):
 
101
 
102
  model_name = model_name.strip().replace(" ", "_")
103
 
 
 
104
  def progress_callback(value, desc):
105
  progress(value, desc=desc)
106