Spaces:
Build error
Build 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
|