Approximetal commited on
Commit
6e31617
·
verified ·
1 Parent(s): 8b982fa

Update inference_gradio.py

Browse files
Files changed (1) hide show
  1. inference_gradio.py +1 -1
inference_gradio.py CHANGED
@@ -46,7 +46,7 @@ class UVR5:
46
  # Keep paths as strings; actual model is loaded lazily.
47
  self.model_dir = str(model_dir)
48
  self.code_dir = str(code_dir)
49
- self.model = None
50
  self.device = "cpu"
51
 
52
  def load_model(self, device: str = "cpu"):
 
46
  # Keep paths as strings; actual model is loaded lazily.
47
  self.model_dir = str(model_dir)
48
  self.code_dir = str(code_dir)
49
+ self.model = self.load_model()
50
  self.device = "cpu"
51
 
52
  def load_model(self, device: str = "cpu"):