Spaces:
Running
Running
fix temp file location
Browse files
app.py
CHANGED
|
@@ -155,7 +155,9 @@ def create_3d_obj(rgb_image, raw_depth, image_path, depth=10, z_scale=200):
|
|
| 155 |
mesh_crop = mesh.crop(bbox)
|
| 156 |
|
| 157 |
# Save the mesh as a GLTF file
|
| 158 |
-
|
|
|
|
|
|
|
| 159 |
o3d.io.write_triangle_mesh(gltf_path, mesh_crop, write_triangle_uvs=True)
|
| 160 |
return gltf_path
|
| 161 |
|
|
@@ -188,7 +190,7 @@ z_scale_slider = gr.Slider(
|
|
| 188 |
examples = [["examples/" + img] for img in os.listdir("examples/")]
|
| 189 |
|
| 190 |
process_image.zerogpu = True
|
| 191 |
-
gr.set_static_paths(paths=["models/","examples/"])
|
| 192 |
iface = gr.Interface(
|
| 193 |
fn=process_image,
|
| 194 |
inputs=[
|
|
@@ -209,8 +211,9 @@ iface = gr.Interface(
|
|
| 209 |
allow_flagging="never",
|
| 210 |
cache_examples=False,
|
| 211 |
delete_cache=(86400,86400),
|
| 212 |
-
theme="Surn/Beeuty"
|
|
|
|
| 213 |
)
|
| 214 |
|
| 215 |
if __name__ == "__main__":
|
| 216 |
-
iface.launch(debug=True, show_api=False, favicon_path="./favicon.ico")
|
|
|
|
| 155 |
mesh_crop = mesh.crop(bbox)
|
| 156 |
|
| 157 |
# Save the mesh as a GLTF file
|
| 158 |
+
temp_dir = Path.cwd() / "models"
|
| 159 |
+
temp_dir.mkdir(exist_ok=True)
|
| 160 |
+
gltf_path = str(temp_dir / f"{image_path.stem}.gltf")
|
| 161 |
o3d.io.write_triangle_mesh(gltf_path, mesh_crop, write_triangle_uvs=True)
|
| 162 |
return gltf_path
|
| 163 |
|
|
|
|
| 190 |
examples = [["examples/" + img] for img in os.listdir("examples/")]
|
| 191 |
|
| 192 |
process_image.zerogpu = True
|
| 193 |
+
#gr.set_static_paths(paths=["models/","examples/"])
|
| 194 |
iface = gr.Interface(
|
| 195 |
fn=process_image,
|
| 196 |
inputs=[
|
|
|
|
| 211 |
allow_flagging="never",
|
| 212 |
cache_examples=False,
|
| 213 |
delete_cache=(86400,86400),
|
| 214 |
+
theme="Surn/Beeuty",
|
| 215 |
+
show_progress = 'full'
|
| 216 |
)
|
| 217 |
|
| 218 |
if __name__ == "__main__":
|
| 219 |
+
iface.launch(debug=True, show_api=False, favicon_path="./favicon.ico", allowed_paths=["models/","examples/"])
|