Update app.py

app.py

@@ -11,7 +11,7 @@ MODEL_CACHE = {}
 for id, model in MODELS.items():
     print(f"Loading model {model}...")
     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
+    MODEL_CACHE[model].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}")
 
 @spaces.GPU