Update app.py

app.py

@@ -12,31 +12,15 @@ model_id = "black-forest-labs/FLUX.1-schnell"
 lora_repo = None
 lora_path = None
 
-def load_lora(repo_id, subfolder=""):
+def load_lora(repo_id, style):
     global lora_repo, lora_path
     try:
-        # Essayer plusieurs noms de fichiers courants
-        possible_files = [
-            "lora.safetensors",
-            "flux_lora.safetensors", 
-            "flux-lora.safetensors",
-            "pytorch_lora_weights.safetensors"
-        ]
+        # Construire le nom du fichier basé sur le style
+        filename = f"{style}_lora.safetensors"
         
-        # Si un subfolder est spécifié (ex: "anime")
-        kwargs = {"repo_id": repo_id}
-        if subfolder:
-            kwargs["subfolder"] = subfolder
-        
-        for filename in possible_files:
-            try:
-                lora_path = hf_hub_download(filename=filename, **kwargs)
-                lora_repo = repo_id
-                return f"✅ LoRA chargé : {repo_id}/{subfolder if subfolder else ''} ({filename})"
-            except:
-                continue
-        
-        return f"❌ Aucun fichier LoRA trouvé. Essayez avec un chemin spécifique."
+        lora_path = hf_hub_download(repo_id=repo_id, filename=filename)
+        lora_repo = repo_id
+        return f"✅ LoRA chargé : {repo_id} ({filename})"
     except Exception as e:
         return f"❌ Erreur LoRA : {str(e)}"