Hugging Face's logo

Spaces:
gstranger
/
diffusion_models
Sleeping

App Files Community
George Krupenchenkov commited on
Commit
1058bb6
·
1 Parent(s): 0135480

add hw6 markups

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -106,7 +106,8 @@ def infer(
106
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny",
107
  torch_dtype=torch_dtype, cache_dir="./models_cache")
108
 
109
- else control_mode == "Line Art":
 
110
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_lineart",
111
  torch_dtype=torch_dtype, cache_dir="./models_cache")
112
 
 
106
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny",
107
  torch_dtype=torch_dtype, cache_dir="./models_cache")
108
 
109
+ else:
110
+ control_mode == "Line Art"
111
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_lineart",
112
  torch_dtype=torch_dtype, cache_dir="./models_cache")
113