Spaces:

jbilcke-hf
/

text-to-map

Paused

jbilcke-hf HF Staff commited on Jan 26

Commit

05b82f0

verified ·

1 Parent(s): 533d2f8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ 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."""

 flux_pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
 flux_pipe.load_lora_weights(adapter_id)
+flux_pipe = flux_pipe.to(device)
 def generate_and_process_map(prompt: str) -> str | None:
     """Generate satellite image from prompt and convert to 3D model."""