Spaces:
Running
on
Zero
Running
on
Zero
Commit
Β·
5269c48
1
Parent(s):
4b7cf01
Refactor UI layout in create_app function for improved organization and clarity; update typing hints in Asset class for better type safety.
Browse files- app.py +3 -1
- src/data/asset.py +2 -2
app.py
CHANGED
|
@@ -489,9 +489,11 @@ def create_app():
|
|
| 489 |
scale=4,
|
| 490 |
)
|
| 491 |
random_btn = gr.Button("π Random Seed", variant="secondary", scale=1)
|
|
|
|
| 492 |
pipeline_btn = gr.Button("π― Start Complete Pipeline", variant="primary", size="lg")
|
| 493 |
|
| 494 |
-
|
|
|
|
| 495 |
|
| 496 |
random_btn.click(
|
| 497 |
fn=lambda: int(torch.randint(0, 100000, (1,)).item()),
|
|
|
|
| 489 |
scale=4,
|
| 490 |
)
|
| 491 |
random_btn = gr.Button("π Random Seed", variant="secondary", scale=1)
|
| 492 |
+
|
| 493 |
pipeline_btn = gr.Button("π― Start Complete Pipeline", variant="primary", size="lg")
|
| 494 |
|
| 495 |
+
with gr.Column(scale=1):
|
| 496 |
+
pipeline_skeleton_out = gr.File(label="Final Rigged Model")
|
| 497 |
|
| 498 |
random_btn.click(
|
| 499 |
fn=lambda: int(torch.randint(0, 100000, (1,)).item()),
|
src/data/asset.py
CHANGED
|
@@ -3,7 +3,7 @@ from dataclasses import dataclass
|
|
| 3 |
import numpy as np
|
| 4 |
from numpy import ndarray
|
| 5 |
|
| 6 |
-
from typing import Dict, Union, List, Tuple
|
| 7 |
|
| 8 |
from .order import Order
|
| 9 |
from .raw_data import RawData
|
|
@@ -82,7 +82,7 @@ class Asset(Exporter):
|
|
| 82 |
# pose matrix for skinning loss calculation, shape (J, 4, 4)
|
| 83 |
pose_matrix: Union[ndarray, None]=None
|
| 84 |
|
| 85 |
-
meta: Union[Dict[str,
|
| 86 |
|
| 87 |
@property
|
| 88 |
def N(self):
|
|
|
|
| 3 |
import numpy as np
|
| 4 |
from numpy import ndarray
|
| 5 |
|
| 6 |
+
from typing import Dict, Union, List, Tuple, Any
|
| 7 |
|
| 8 |
from .order import Order
|
| 9 |
from .raw_data import RawData
|
|
|
|
| 82 |
# pose matrix for skinning loss calculation, shape (J, 4, 4)
|
| 83 |
pose_matrix: Union[ndarray, None]=None
|
| 84 |
|
| 85 |
+
meta: Union[Dict[str, Any], None]=None
|
| 86 |
|
| 87 |
@property
|
| 88 |
def N(self):
|