Update app.py

app.py

@@ -185,9 +185,13 @@ def get_huggingface_safetensors(link):
     print(f"Base model: {base_model}")
 
     # Validate model type
-    if base_model not in {"black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-schnell"}:
+    acceptable_models = {"black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-schnell"}
+    
+    models_to_check = base_model if isinstance(base_model, list) else [base_model]
+    
+    if not any(model in acceptable_models for model in models_to_check):
         raise Exception("Not a FLUX LoRA!")
-
+        
     # Extract image and trigger word
     print("Before trying to get image")
     image_path = model_card.data.get("widget", [{}])[0].get("output", {}).get("url", None)