Spaces:
Sleeping
Sleeping
Update demo_support/utils.py
Browse files- 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 |
-
|
110 |
-
|
111 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
|