Charbel Malo
commited on
Commit
•
9d5d0ed
1
Parent(s):
e22f41b
Update app.py
Browse files
app.py
CHANGED
|
@@ -157,10 +157,8 @@ def load_nsfw_detector_model(path="./assets/pretrained_models/open-nsfw.onnx"):
|
|
| 157 |
NSFW_DETECTOR = NSFWChecker(model_path=path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
| 158 |
|
| 159 |
|
| 160 |
-
@spaces.GPU(enable_queue=True)
|
| 161 |
-
load_face_analyser_model()
|
| 162 |
|
| 163 |
-
|
| 164 |
load_face_swapper_model()
|
| 165 |
|
| 166 |
## ------------------------------ MAIN PROCESS ------------------------------
|
|
|
|
| 157 |
NSFW_DETECTOR = NSFWChecker(model_path=path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
| 158 |
|
| 159 |
|
|
|
|
|
|
|
| 160 |
|
| 161 |
+
load_face_analyser_model()
|
| 162 |
load_face_swapper_model()
|
| 163 |
|
| 164 |
## ------------------------------ MAIN PROCESS ------------------------------
|