Update app.py

app.py

@@ -124,8 +124,8 @@ def infer(
 
     # print('ip_adapter_mode.value = ', ip_adapter_mode.value)
     # print('control_mode.value = ', control_mode.value) 
-    print('ip_adapter_mode.value = ', ip_adapter_mode)
-    print('control_mode.value = ', control_mode) 
+    print('ip_adapter_mode = ', ip_adapter_mode)
+    print('control_mode = ', control_mode) 
     
     # Генерация изображений с Ip_Adapter ------------------------------------------------------------------------------------------------------------------
     if use_ip_adapter: #and ip_source_image is not None and ip_adapter_image is not None:
@@ -151,9 +151,10 @@ def infer(
 
         # #ip_adapter_mode = "pose_estimation" # Режим работы Ip_Adapter 
 
-        if ip_adapter_mode.value == "pose_estimation":
+        #if ip_adapter_mode.value == "pose_estimation":
+        if ip_adapter_mode == "pose_estimation":    
 
-            print('ip_adapter_mode.value = ', ip_adapter_mode.value)
+            print('ip_adapter_mode = ', ip_adapter_mode)
             
             # Инициализация ControlNet
             controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch_dtype)
@@ -240,9 +241,10 @@ def infer(
 
             # #ip_adapter_mode = "edge_detection" # Режим работы Ip_Adapter 
 
-            if ip_adapter_mode.value == "edge_detection":
+            #if ip_adapter_mode.value == "edge_detection":
+            if ip_adapter_mode == "edge_detection":    
 
-                print('ip_adapter_mode.value = ', ip_adapter_mode.value)
+                print('ip_adapter_mode = ', ip_adapter_mode)
 
                 # Инициализация ControlNet
                 controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
@@ -346,9 +348,10 @@ def infer(
                 
                 # #ip_adapter_mode = "depth_map" # Режим работы Ip_Adapter 
 
-                if ip_adapter_mode.value == "depth_map":
+                #if ip_adapter_mode.value == "depth_map":
+                if ip_adapter_mode == "depth_map":    
 
-                    print('ip_adapter_mode.value = ', ip_adapter_mode.value)
+                    print('ip_adapter_mode = ', ip_adapter_mode)
 
                    
                     # Инициализация ControlNet