Update app.py

app.py

@@ -377,15 +377,13 @@ def infer(
         print("❌ Invalid key.")
         return None
     config = LORA_CONFIG[lora_name]
-    
-    if config["type"] == "style":
-        if style_image is None:
-            raise gr.Error("Style Transfer LoRA requires a Style Reference Image.")
-        image_for_pipeline = style_image
-    else:  # 'edit' or 'base'
-        if input_image is None:
-            raise gr.Error("This LoRA requires an Input Image.")
-        image_for_pipeline = input_image
+
+    default_path = os.path.join(os.path.dirname(__file__), "1.jpg")
+    if os.path.exists(default_path):
+        image_for_pipeline = [Image.open(default_path).convert("RGB")]
+        print("Loaded default image: 1.jpg")
+    else:
+        raise gr.Error("No input images and '1.jpg' not found in app directory.")
 
     if not prompt and config["prompt_template"] != "change the face to face segmentation mask":
         raise gr.Error("A text prompt is required for this LoRA.")