winfred2027 commited on
Commit
67148b3
·
verified ·
1 Parent(s): 104f14f

Update demo_support/utils.py

Browse files
Files changed (1) hide show
  1. demo_support/utils.py +13 -3
demo_support/utils.py CHANGED
@@ -106,9 +106,19 @@ def input_3d_shape(key=None):
106
  model_key = key + "_model"
107
  npy_key = key + "_npy"
108
  swap_key = key + "_swap"
109
- objaid = st.sidebar.text_input("Enter an Objaverse ID", key=objaid_key)
110
- model = st.sidebar.file_uploader("Or upload a model (.glb/.obj/.ply)", key=model_key)
111
- npy = st.sidebar.file_uploader("Or upload a point cloud numpy array (.npy of Nx3 XYZ or Nx6 XYZRGB)", key=npy_key)
 
 
 
 
 
 
 
 
 
 
112
  swap_yz_axes = st.sidebar.radio("Gravity", ["Y is up (for most Objaverse shapes)", "Z is up"], key=swap_key) == "Z is up"
113
  f32 = numpy.float32
114
 
 
106
  model_key = key + "_model"
107
  npy_key = key + "_npy"
108
  swap_key = key + "_swap"
109
+ pc_mode = st.sidebar.selectbox(
110
+ 'Choose the input for Point Cloud',
111
+ ("an Objaverse ID", "a Model(.glb/.obj/.ply)", "a Numpy Array(.npy)")
112
+ )
113
+ if pc_mode = "an Objaverse ID":
114
+ objaid = st.sidebar.text_input("Enter an Objaverse ID", key=objaid_key)
115
+ model = npy = None
116
+ elif pc_mode = "an Objaverse ID":
117
+ model = st.sidebar.file_uploader("Upload a model (.glb/.obj/.ply)", key=model_key)
118
+ objaid = npy = None
119
+ elif pc_mode = "an Objaverse ID":
120
+ npy = st.sidebar.file_uploader("Upload a numpy array (.npy of Nx3 XYZ or Nx6 XYZRGB)", key=npy_key)
121
+ objaid = model = None
122
  swap_yz_axes = st.sidebar.radio("Gravity", ["Y is up (for most Objaverse shapes)", "Z is up"], key=swap_key) == "Z is up"
123
  f32 = numpy.float32
124