jbilcke-hf HF Staff commited on
Commit
426ec64
·
verified ·
1 Parent(s): ced5e36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,4 +1,3 @@
1
- # text_to_map_app.py
2
  import os
3
  import tempfile
4
  import torch
@@ -179,7 +178,7 @@ class GPUSatelliteModelGenerator:
179
  final_height = size_factor * shadow_factor
180
  height_map[building_mask] = final_height
181
 
182
- return height_map.get() * 0.15
183
 
184
  def generate_mesh_gpu(self, height_map, texture_img):
185
  """Generate 3D mesh using GPU-accelerated calculations"""
@@ -255,7 +254,7 @@ def generate_and_process_map(prompt: str) -> str | None:
255
  prompt=prompt,
256
  width=width,
257
  height=height,
258
- num_inference_steps=30,
259
  generator=generator,
260
  guidance_scale=7.5
261
  ).images[0]
 
 
1
  import os
2
  import tempfile
3
  import torch
 
178
  final_height = size_factor * shadow_factor
179
  height_map[building_mask] = final_height
180
 
181
+ return height_map.get() * 0.25
182
 
183
  def generate_mesh_gpu(self, height_map, texture_img):
184
  """Generate 3D mesh using GPU-accelerated calculations"""
 
254
  prompt=prompt,
255
  width=width,
256
  height=height,
257
+ num_inference_steps=25,
258
  generator=generator,
259
  guidance_scale=7.5
260
  ).images[0]