Spaces:
Running
on
Zero
Running
on
Zero
fix: planes to image
Browse files
app.py
CHANGED
@@ -227,16 +227,16 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
|
|
227 |
return None, None
|
228 |
|
229 |
def step_1_generate_planes(image):
|
230 |
-
|
231 |
-
return
|
232 |
|
233 |
def step_2_generate_obj(image):
|
234 |
-
|
235 |
-
return
|
236 |
|
237 |
def step_3_generate_video(image):
|
238 |
-
|
239 |
-
return
|
240 |
|
241 |
# 从 assets 文件夹中设置示例文件,并限制最多读取 10 个文件
|
242 |
example_folder = "assets"
|
|
|
227 |
return None, None
|
228 |
|
229 |
def step_1_generate_planes(image):
|
230 |
+
planes, _ = generate_mesh(image)
|
231 |
+
return planes
|
232 |
|
233 |
def step_2_generate_obj(image):
|
234 |
+
planes, mesh_path = generate_mesh(image, export_mesh=True)
|
235 |
+
return planes, mesh_path, mesh_path
|
236 |
|
237 |
def step_3_generate_video(image):
|
238 |
+
planes, video_path = generate_mesh(image, export_video=True)
|
239 |
+
return planes, video_path
|
240 |
|
241 |
# 从 assets 文件夹中设置示例文件,并限制最多读取 10 个文件
|
242 |
example_folder = "assets"
|