hassiahk commited on
Commit
3ea7c11
1 Parent(s): 4af1950

Added latest models for Hotdog and Drums

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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": "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu",
22
- "NERF_FILE_ID": "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu",
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
  }