Lifeinhockey commited on
Commit
378b068
·
verified ·
1 Parent(s): 8338f41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -387,9 +387,9 @@ def infer(
387
 
388
  # #control_mode = "pose_estimation" # Режим работы ControlNet
389
 
390
- print('control_mode = ', control_mode.choices)
391
 
392
- if control_mode.choices == "pose_estimation":
393
 
394
  # Инициализация ControlNet
395
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch_dtype)
 
387
 
388
  # #control_mode = "pose_estimation" # Режим работы ControlNet
389
 
390
+ print('control_mode = ', control_mode.value)
391
 
392
+ if control_mode.value == "pose_estimation":
393
 
394
  # Инициализация ControlNet
395
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch_dtype)