Spaces:
Runtime error
Runtime error
root
commited on
Commit
·
49f2d34
1
Parent(s):
649468d
update
Browse files- app.py +4 -4
- roop/predictor.py +1 -1
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import roop.globals
|
|
| 6 |
from roop.core import (
|
| 7 |
start,
|
| 8 |
decode_execution_providers,
|
| 9 |
-
suggest_max_memory,
|
| 10 |
suggest_execution_threads,
|
| 11 |
)
|
| 12 |
from roop.processors.frame.core import get_frame_processors_modules
|
|
@@ -45,9 +45,9 @@ def swap_face(source_file, target_file,doFaceEnhancer):
|
|
| 45 |
roop.globals.many_faces = False
|
| 46 |
roop.globals.video_encoder = "libx264"
|
| 47 |
roop.globals.video_quality = 18
|
| 48 |
-
roop.globals.max_memory = suggest_max_memory()
|
| 49 |
-
roop.globals.execution_providers = decode_execution_providers(["
|
| 50 |
-
roop.globals.execution_threads = suggest_execution_threads()
|
| 51 |
|
| 52 |
print(
|
| 53 |
"start process",
|
|
|
|
| 6 |
from roop.core import (
|
| 7 |
start,
|
| 8 |
decode_execution_providers,
|
| 9 |
+
#suggest_max_memory,
|
| 10 |
suggest_execution_threads,
|
| 11 |
)
|
| 12 |
from roop.processors.frame.core import get_frame_processors_modules
|
|
|
|
| 45 |
roop.globals.many_faces = False
|
| 46 |
roop.globals.video_encoder = "libx264"
|
| 47 |
roop.globals.video_quality = 18
|
| 48 |
+
#roop.globals.max_memory = suggest_max_memory()
|
| 49 |
+
roop.globals.execution_providers = decode_execution_providers(["cpu"])
|
| 50 |
+
#roop.globals.execution_threads = suggest_execution_threads()
|
| 51 |
|
| 52 |
print(
|
| 53 |
"start process",
|
roop/predictor.py
CHANGED
|
@@ -8,7 +8,7 @@ from roop.typing import Frame
|
|
| 8 |
|
| 9 |
PREDICTOR = None
|
| 10 |
THREAD_LOCK = threading.Lock()
|
| 11 |
-
MAX_PROBABILITY =
|
| 12 |
|
| 13 |
|
| 14 |
def get_predictor() -> Model:
|
|
|
|
| 8 |
|
| 9 |
PREDICTOR = None
|
| 10 |
THREAD_LOCK = threading.Lock()
|
| 11 |
+
MAX_PROBABILITY = 1
|
| 12 |
|
| 13 |
|
| 14 |
def get_predictor() -> Model:
|