Spaces:
Build error
Build error
Added latest models for Hotdog and Drums
Browse files- app.py +1 -1
- config.json +8 -2
app.py
CHANGED
@@ -18,7 +18,7 @@ MODEL_DIR = "models"
|
|
18 |
|
19 |
|
20 |
def select_model():
|
21 |
-
obj_select = st.selectbox("Select a Scene", ("Mic", "Chair", "Lego", "Ship", "Hotdog"))
|
22 |
DIET_NERF_MODEL_NAME = cfg[obj_select]["DIET_NERF_MODEL_NAME"]
|
23 |
DIET_NERF_FILE_ID = cfg[obj_select]["DIET_NERF_FILE_ID"]
|
24 |
|
|
|
18 |
|
19 |
|
20 |
def select_model():
|
21 |
+
obj_select = st.selectbox("Select a Scene", ("Mic", "Chair", "Lego", "Ship", "Hotdog", "Drums"))
|
22 |
DIET_NERF_MODEL_NAME = cfg[obj_select]["DIET_NERF_MODEL_NAME"]
|
23 |
DIET_NERF_FILE_ID = cfg[obj_select]["DIET_NERF_FILE_ID"]
|
24 |
|
config.json
CHANGED
@@ -18,8 +18,8 @@
|
|
18 |
"NERF_MODEL_NAME": "nerf_ship"
|
19 |
},
|
20 |
"Hotdog": {
|
21 |
-
"DIET_NERF_FILE_ID": "
|
22 |
-
"NERF_FILE_ID": "
|
23 |
"DIET_NERF_MODEL_NAME": "diet_nerf_hotdog",
|
24 |
"NERF_MODEL_NAME": "nerf_hotdog"
|
25 |
},
|
@@ -28,5 +28,11 @@
|
|
28 |
"NERF_FILE_ID": "1JK3vu2F9xnOuhPjYVflMkPp0BcAc4w_9",
|
29 |
"DIET_NERF_MODEL_NAME": "diet_nerf_mic",
|
30 |
"NERF_MODEL_NAME": "nerf_mic"
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
}
|
32 |
}
|
|
|
18 |
"NERF_MODEL_NAME": "nerf_ship"
|
19 |
},
|
20 |
"Hotdog": {
|
21 |
+
"DIET_NERF_FILE_ID": "1I_Gy2yeaaQrhX3IZ-jTpgHMP1eMp2Bxj",
|
22 |
+
"NERF_FILE_ID": "1dlevr64cEW83Ka7v1tWwzoycI3K95lPy",
|
23 |
"DIET_NERF_MODEL_NAME": "diet_nerf_hotdog",
|
24 |
"NERF_MODEL_NAME": "nerf_hotdog"
|
25 |
},
|
|
|
28 |
"NERF_FILE_ID": "1JK3vu2F9xnOuhPjYVflMkPp0BcAc4w_9",
|
29 |
"DIET_NERF_MODEL_NAME": "diet_nerf_mic",
|
30 |
"NERF_MODEL_NAME": "nerf_mic"
|
31 |
+
},
|
32 |
+
"Drums": {
|
33 |
+
"DIET_NERF_FILE_ID": "11qpHv-LeiKtLcO3LwBsM9A0rz23IZdjS",
|
34 |
+
"NERF_FILE_ID": "1GVgD3WpCKOmWMLqTU5QMMcoXdPTq5vQk",
|
35 |
+
"DIET_NERF_MODEL_NAME": "diet_nerf_mic",
|
36 |
+
"NERF_MODEL_NAME": "nerf_mic"
|
37 |
}
|
38 |
}
|