MohamedRashad commited on
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
Files changed (2) hide show
  1. app.py +3 -1
  2. 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
- pipeline_skeleton_out = gr.File(label="Final Rigged Model")
 
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, ...], None]=None
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):