Update app.py

app.py

@@ -8,7 +8,7 @@ MODELS = {
     'OpenFLUX.1': 'ostris/OpenFLUX.1',
 }
 MODEL_CACHE = {}
-for id, model in MODELS:
+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)
     pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power