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 |
}
|