update

app.py

@@ -9,6 +9,8 @@
 """
 
 HuggingFace = True
+update_model_id = 0
+base_download_path = "downloaded"
 
 if HuggingFace is False:
     from dotenv import load_dotenv
@@ -24,6 +26,26 @@ import os
 
 query_params = st.experimental_get_query_params()
 
+#------------
+if update_model_id is not None:
+    if HuggingFace is False:
+        model_name = os.getenv("m{}_name".format(update_model_id))
+        model_extname = os.getenv("m{}_type".format(update_model_id))
+
+    else:
+        model_name = st.secrets["m{}_name".format(update_model_id)]
+        model_extname = st.secrets["m{}_type".format(update_model_id)]
+
+    path_model = os.path.join(base_download_path, model_name + model_extname)
+    if os.path.exists(path_model):
+        try:
+            os.remove(path_model)
+        except:
+            print("Cannot remove", path_model)
+            continue
+
+#-------------
+
 qmodel = 'crowded_human'
 if 'model' in query_params:
     qmodel = query_params['model'][0]
@@ -37,7 +59,6 @@ st.set_page_config(
     )
 
 
-base_download_path = "downloaded"
 #------------------------
 if not os.path.exists(base_download_path):
     os.makedirs(base_download_path)