Update app.py

app.py

@@ -12,11 +12,12 @@ from script import SatelliteModelGenerator
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16
 
-# Initialize FLUX model for satellite imagery
-flux_pipe = DiffusionPipeline.from_pretrained(
-    "jbilcke-hf/flux-satellite",
-    torch_dtype=dtype
-).to(device)
+repo_id = "black-forest-labs/FLUX.1-dev"
+adapter_id = "jbilcke-hf/flux-satellite"
+
+flux_pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
+flux_pipe.load_lora_weights(adapter_id)
+flux_pipe = pipeline.to(device)
 
 def generate_and_process_map(prompt: str) -> str | None:
     """Generate satellite image from prompt and convert to 3D model."""