Approximetal commited on
Commit
8c13411
·
verified ·
1 Parent(s): b742966

Update inference_gradio.py

Browse files
Files changed (1) hide show
  1. inference_gradio.py +1 -3
inference_gradio.py CHANGED
@@ -75,9 +75,7 @@ class UVR5:
75
  )
76
 
77
  uvr5_model = Inference(model_data, "cpu")
78
- uvr5_model.load_model(model_path, 1)
79
-
80
- self.model = uvr5_model.load_model(device="cpu")
81
  self.device = "cpu"
82
  return self.model
83
 
 
75
  )
76
 
77
  uvr5_model = Inference(model_data, "cpu")
78
+ self.model = uvr5_model.load_model(model_path, 1, device="cpu")
 
 
79
  self.device = "cpu"
80
  return self.model
81