add hw6 markups

app.py

@@ -97,12 +97,16 @@ def infer(
             'height': height,
             'generator': generator,
             }
+    
+    print("in infer 1")
 
     if controlnet_enable:
+        print("in controlnet_enable")
         if control_mode == "canny":
             controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny", 
                                                          torch_dtype=torch_dtype, cache_dir="./models_cache")
-        elif control_mode == "Line Art":
+            
+        else control_mode == "Line Art":
             controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_lineart", 
                                                          torch_dtype=torch_dtype, cache_dir="./models_cache")
             
@@ -116,6 +120,7 @@ def infer(
 
 
     else:
+        print("in infer 2")
         pipe = StableDiffusionPipeline.from_pretrained(model_id, 
                                                        torch_dtype=torch_dtype, 
                                                        safety_checker=None) #.to(device)