Update app.py

app.py

@@ -557,8 +557,10 @@ def get_huggingface_safetensors(link):
             base_model = model_card.data.get("base_model")
             print(base_model)
       
+            #Allows Both
             if((base_model != "black-forest-labs/FLUX.1-dev") and (base_model != "black-forest-labs/FLUX.1-schnell")):
                 raise Exception("Flux LoRA Not Found!")
+                
             # Only allow "black-forest-labs/FLUX.1-dev"
             #if base_model != "black-forest-labs/FLUX.1-dev":
                 #raise Exception("Only FLUX.1-dev is supported, other LoRA models are not allowed!")