Update app.py

app.py

@@ -15,12 +15,13 @@ def init_pipeline():
     pipe = FluxPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
     )
-    pipe = pipe.to("cuda")
+    #pipe = pipe.to("cuda")
     pipe.load_lora_weights(
         "Yuanshi/OminiControl",
         weight_name=f"omini/subject_512.safetensors",
         adapter_name="subject",
     )
+    pipe.enable_model_cpu_offload() 
 
 
 def process_image_and_text(image, text):