radames HF staff commited on
Commit
cfd6e34
1 Parent(s): 9b24eb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import retinaface
11
  from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
12
  from diffusers import UniPCMultistepScheduler
13
 
14
- from spiga.inference.config import ModelConfig
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