Update app.py

app.py

@@ -45,7 +45,7 @@ class ImageProcessor:
 
 # Initialize the image processor
 base_path = "black-forest-labs/FLUX.1-dev"    
-lora_base_path = "EasyControl/models"
+lora_base_path = "./models"
 style_lora_base_path = "Shakker-Labs"
 processor = ImageProcessor(base_path)
 
@@ -54,18 +54,10 @@ def single_condition_generate_image(prompt, subject_img, spatial_img, height, wi
     # Set the control type
     if control_type == "subject":
         lora_path = os.path.join(lora_base_path, "subject.safetensors")
-    elif control_type == "depth":
-        lora_path = os.path.join(lora_base_path, "depth.safetensors")
-    elif control_type == "seg":
-        lora_path = os.path.join(lora_base_path, "seg.safetensors")
     elif control_type == "pose":
         lora_path = os.path.join(lora_base_path, "pose.safetensors")
     elif control_type == "inpainting":
         lora_path = os.path.join(lora_base_path, "inpainting.safetensors")
-    elif control_type == "hedsketch":
-        lora_path = os.path.join(lora_base_path, "hedsketch.safetensors")
-    elif control_type == "canny":
-        lora_path = os.path.join(lora_base_path, "canny.safetensors")
     set_single_lora(processor.pipe.transformer, lora_path, lora_weights=[1], cond_size=512)
     
     # Set the style LoRA
@@ -104,12 +96,12 @@ def multi_condition_generate_image(prompt, subject_img, spatial_img, height, wid
     return image
 
 # Define the Gradio interface components
-control_types = ["subject", "depth", "pose", "inpainting", "hedsketch", "seg", "canny"]
+control_types = ["subject", "pose", "inpainting"]
 style_loras = ["Simple_Sketch", "Text_Poster", "Vector_Style", "None"]
 
 # Example data
 single_examples = [
-    ["A SKS in the library", Image.open("./test_imgs/subject1.png"), None, 1024, 1024, 5, "subject", None],
+    ["A SKS in the library", Image.open("./test_imgs/subject1.png"), None, 1024, 1024, 5, "subject", "None"],
     ["In a picturesque village, a narrow cobblestone street with rustic stone buildings, colorful blinds, and lush green spaces, a cartoon man drawn with simple lines and solid colors stands in the foreground, wearing a red shirt, beige work pants, and brown shoes, carrying a strap on his shoulder. The scene features warm and enticing colors, a pleasant fusion of nature and architecture, and the camera's perspective on the street clearly shows the charming and quaint environment., Integrating elements of reality and cartoon.", None, Image.open("./test_imgs/spatial1.png"), 1024, 1024, 1, "pose", "Vector_Style"],
 ]
 multi_examples = [