Spaces:
Running
on
Zero
Running
on
Zero
Update inference_gradio.py
Browse files- 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 =
|
| 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"):
|