Added updated models

app.py

@@ -19,7 +19,7 @@ with open("config.json") as f:
 
 
 def select_model():
-    obj_select = st.selectbox("Select a scene", ("Chair", "Lego", "Ship", "Hotdog"))
+    obj_select = st.selectbox("Select a scene", ("Mic", "Chair", "Lego", "Ship", "Hotdog"))
     # if obj_select == "Chair":
     #     FILE_ID = "17dj0pQieo94TozFv-noSBkXebduij1aM"
     #     MODEL_DIR = "models"
@@ -54,7 +54,7 @@ st.markdown("")
 MODEL_DIR, MODEL_NAME, FILE_ID = select_model()
 
 
-@st.cache
+@st.cache(show_spinner=False)
 def download_model():
     os.makedirs(MODEL_DIR, exist_ok=True)
     _model_path = os.path.join(MODEL_DIR, MODEL_NAME)
@@ -87,7 +87,7 @@ st.sidebar.markdown(
 }
 </style>
 <p class="aligncenter">
-    <img src="https://user-images.githubusercontent.com/77657524/126361638-4aad58e8-4efb-4fc5-bf78-f53d03799e1e.png" width="410" height="400"/>
+    <img src="https://user-images.githubusercontent.com/77657524/126361638-4aad58e8-4efb-4fc5-bf78-f53d03799e1e.png" width="420" height="400"/>
 </p>
 """,
     unsafe_allow_html=True,

config.json

@@ -5,12 +5,12 @@
         "NERF_MODEL_NAME": "nerf_chair"
     },
     "Lego": {
-        "FILE_ID": "1D9I-qIVMPaxuCHfUWPWMHaoLYtAmCjwI",
+        "FILE_ID": "1UaUKgShJTKp5zNQdXvVhkyxgU-2YU3Ay",
         "DIET_NERF_MODEL_NAME": "diet_nerf_lego",
         "NERF_MODEL_NAME": "nerf_lego"
     },
     "Ship": {
-        "FILE_ID": "14ZeJ86ETQr8dtu6CFoxU-ifvniHKo_Dt",
+        "FILE_ID": "1Eb4eZ9vn6FcbqLIaBoW-6NlyWr-y0eTW",
         "DIET_NERF_MODEL_NAME": "diet_nerf_ship",
         "NERF_MODEL_NAME": "nerf_ship"
     },
@@ -18,5 +18,10 @@
         "FILE_ID": "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu",
         "DIET_NERF_MODEL_NAME": "diet_nerf_hotdog",
         "NERF_MODEL_NAME": "nerf_hotdog"
+    },
+    "Mic": {
+        "FILE_ID": "1vuyCAY8YKLo7OSlTpLJgMem47gJc1W2S",
+        "DIET_NERF_MODEL_NAME": "diet_nerf_mic",
+        "NERF_MODEL_NAME": "nerf_mic"
     }
 }