Update app.py

app.py

@@ -42,9 +42,9 @@ class ImageProcessor:
         return image
 
 # Initialize the image processor
-base_path = "/opt/liblibai-models/model-weights/black-forest-labs/FLUX.1-dev"    
-lora_base_path = "/opt/liblibai-models/user-workspace/rey/projects/opensource/github/EasyControl/models"
-style_lora_base_path = "/opt/liblibai-models/user-workspace/zhangyuxuan/models/Shakker-Labs"
+base_path = "black-forest-labs/FLUX.1-dev"    
+lora_base_path = "EasyControl/models"
+style_lora_base_path = "Shakker-Labs"
 processor = ImageProcessor(base_path)
 
 # Define the Gradio interface
@@ -72,16 +72,16 @@ def single_condition_generate_image(prompt, subject_img, spatial_img, height, wi
     else:
         if style_lora == "Simple_Sketch":
             processor.pipe.unload_lora_weights()
-            style_lora_path = os.path.join(style_lora_base_path, "FLUX.1-dev-LoRA-Children-Simple-Sketch/pytorch_lora_weights.safetensors")
-            processor.pipe.load_lora_weights(self.lora_path, weight_name="pytorch_lora_weights.safetensors")
+            style_lora_path = os.path.join(style_lora_base_path, "FLUX.1-dev-LoRA-Children-Simple-Sketch")
+            processor.pipe.load_lora_weights(style_lora_path, weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
         if style_lora == "Text_Poster":
             processor.pipe.unload_lora_weights()
-            style_lora_path = os.path.join(style_lora_base_path, "FLUX.1-dev-LoRA-Text-Poster/pytorch_lora_weights.safetensors")
-            processor.pipe.load_lora_weights(self.lora_path, weight_name="pytorch_lora_weights.safetensors")
+            style_lora_path = os.path.join(style_lora_base_path, "FLUX.1-dev-LoRA-Text-Poster")
+            processor.pipe.load_lora_weights(style_lora_path, weight_name="FLUX-dev-lora-Text-Poster.safetensors")
         if style_lora == "Vector_Style":
             processor.pipe.unload_lora_weights()
-            style_lora_path = os.path.join(style_lora_base_path, "FLUX.1-dev-LoRA-Vector-Journey/pytorch_lora_weights.safetensors")
-            processor.pipe.load_lora_weights(style_lora_path, weight_name="pytorch_lora_weights.safetensors")
+            style_lora_path = os.path.join(style_lora_base_path, "FLUX.1-dev-LoRA-Vector-Journey")
+            processor.pipe.load_lora_weights(style_lora_path, weight_name="FLUX-dev-lora-Vector-Journey.safetensors")
 
     # Process the image
     subject_imgs = [subject_img] if subject_img else []
@@ -107,11 +107,11 @@ style_loras = ["Simple_Sketch", "Text_Poster", "Vector_Style", "None"]
 
 # Example data
 single_examples = [
-    ["A SKS in the library", Image.open("/opt/liblibai-models/user-workspace/zhangyuxuan/project/easycontrol/inference0310/test_imgs/subject3.jpg"), 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("/opt/liblibai-models/user-workspace/zhangyuxuan/project/easycontrol/inference0310/test_imgs/openpose.png"), 1024, 1024, 1, "pose", "Vector_Style"],
+    ["A SKS in the library", Image.open("./test_imgs/subject1.jpg"), 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/openpose.png"), 1024, 1024, 1, "pose", "Vector_Style"],
 ]
 multi_examples = [
-    ["A SKS on the car", Image.open("/opt/liblibai-models/user-workspace/zhangyuxuan/project/easycontrol/code0221/test_imgs/subject/s17.png"), Image.open("/opt/liblibai-models/user-workspace/zhangyuxuan/project/easycontrol/code0221/test_imgs/inpainting.png"), 1024, 1024, 7],
+    ["A SKS on the car", Image.open("./test_imgs/subject2.png"), Image.open("./test_imgs/inpainting.png"), 1024, 1024, 7],
 ]
 
 
@@ -141,7 +141,7 @@ with gr.Blocks() as demo:
             inputs=[prompt, subject_img, spatial_img, height, width, seed, control_type, style_lora],
             outputs=single_output_image,
             fn=single_condition_generate_image,
-            cache_examples=True,  # 缓存示例结果以加快加载速度
+            cache_examples=False,  # 缓存示例结果以加快加载速度
             label="Single Condition Examples"
         )
 
@@ -165,7 +165,7 @@ with gr.Blocks() as demo:
             inputs=[multi_prompt, multi_subject_img, multi_spatial_img, multi_height, multi_width, multi_seed],
             outputs=multi_output_image,
             fn=multi_condition_generate_image,
-            cache_examples=True,  # 缓存示例结果以加快加载速度
+            cache_examples=False,  # 缓存示例结果以加快加载速度
             label="Multi-Condition Examples"
         )
 
@@ -183,4 +183,4 @@ with gr.Blocks() as demo:
     )
 
 # Launch the Gradio app
-demo.queue().launch(server_name='0.0.0.0',server_port=7861)
+demo.queue().launch()