Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ import retinaface
|
|
11 |
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
|
12 |
from diffusers import UniPCMultistepScheduler
|
13 |
|
14 |
-
from
|
15 |
from spiga.inference.framework import SPIGAFramework
|
16 |
import spiga.demo.analyze.track.retinasort.config as cfg
|
17 |
|
@@ -26,7 +26,7 @@ face_detector = retinaface.RetinaFaceDetector(model=config['retina']['model_name
|
|
26 |
extra_features=config['retina']['extra_features'],
|
27 |
cfg_postreat=config['retina']['postreat'])
|
28 |
# Landmark extraction
|
29 |
-
spiga_extractor = SPIGAFramework(ModelConfig("300wpublic"))
|
30 |
|
31 |
uncanny_controlnet = ControlNetModel.from_pretrained(
|
32 |
"multimodalart/uncannyfaces_25K", torch_dtype=torch.float16
|
|
|
11 |
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
|
12 |
from diffusers import UniPCMultistepScheduler
|
13 |
|
14 |
+
from modelconfig import ModelConfig
|
15 |
from spiga.inference.framework import SPIGAFramework
|
16 |
import spiga.demo.analyze.track.retinasort.config as cfg
|
17 |
|
|
|
26 |
extra_features=config['retina']['extra_features'],
|
27 |
cfg_postreat=config['retina']['postreat'])
|
28 |
# Landmark extraction
|
29 |
+
spiga_extractor = SPIGAFramework(ModelConfig("300wpublic", False))
|
30 |
|
31 |
uncanny_controlnet = ControlNetModel.from_pretrained(
|
32 |
"multimodalart/uncannyfaces_25K", torch_dtype=torch.float16
|