Spaces:
Running
on
Zero
Running
on
Zero
even earlier
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ def text_to_3d(
|
|
| 89 |
slat_guidance_strength: float,
|
| 90 |
slat_sampling_steps: int,
|
| 91 |
req: gr.Request,
|
| 92 |
-
) -> Tuple[dict, str
|
| 93 |
"""
|
| 94 |
Convert an text prompt to a 3D model.
|
| 95 |
Args:
|
|
@@ -102,7 +102,6 @@ def text_to_3d(
|
|
| 102 |
Returns:
|
| 103 |
dict: The information of the generated 3D model.
|
| 104 |
str: The path to the video of the 3D model.
|
| 105 |
-
dict: The state of the generated 3D model.
|
| 106 |
"""
|
| 107 |
user_dir = os.path.join(TMP_DIR, str(req.session_hash))
|
| 108 |
os.makedirs(user_dir, exist_ok=True)
|
|
@@ -126,8 +125,7 @@ def text_to_3d(
|
|
| 126 |
imageio.mimsave(video_path, video, fps=15)
|
| 127 |
state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
|
| 128 |
torch.cuda.empty_cache()
|
| 129 |
-
|
| 130 |
-
return state, video_path, state
|
| 131 |
|
| 132 |
|
| 133 |
@spaces.GPU(duration=90)
|
|
|
|
| 89 |
slat_guidance_strength: float,
|
| 90 |
slat_sampling_steps: int,
|
| 91 |
req: gr.Request,
|
| 92 |
+
) -> Tuple[dict, str]:
|
| 93 |
"""
|
| 94 |
Convert an text prompt to a 3D model.
|
| 95 |
Args:
|
|
|
|
| 102 |
Returns:
|
| 103 |
dict: The information of the generated 3D model.
|
| 104 |
str: The path to the video of the 3D model.
|
|
|
|
| 105 |
"""
|
| 106 |
user_dir = os.path.join(TMP_DIR, str(req.session_hash))
|
| 107 |
os.makedirs(user_dir, exist_ok=True)
|
|
|
|
| 125 |
imageio.mimsave(video_path, video, fps=15)
|
| 126 |
state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
|
| 127 |
torch.cuda.empty_cache()
|
| 128 |
+
return state, video_path
|
|
|
|
| 129 |
|
| 130 |
|
| 131 |
@spaces.GPU(duration=90)
|