Update app.py

app.py

@@ -10,7 +10,7 @@ MODELS = {
 MODEL_CACHE = {}
 for id, model in MODELS.items():
     print(f"Loading model {model}...")
-    MODEL_CACHE[model] = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+    MODEL_CACHE[model] = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
     pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
     print(f"Loaded model {model}")