update

app.py

@@ -7,8 +7,12 @@
    @Description:
 -------------------------------------------------
 """
-#from dotenv import load_dotenv
-#load_dotenv()
+
+HuggingFace = True
+
+if HuggingFace is False:
+    from dotenv import load_dotenv
+    load_dotenv()
 
 from pathlib import Path
 from PIL import Image
@@ -32,21 +36,27 @@ base_download_path = "downloaded"
 if not os.path.exists(base_download_path):
     os.makedirs(base_download_path)
 
-#model_count = int(os.getenv("model_count"))
-model_count = int(st.secrets["model_count"])
+if HuggingFace is False:
+    model_count = int(os.getenv("model_count"))
+
+else:
+    model_count = int(st.secrets["model_count"])
 
 model_info = {}
 models_list = []
 for i in range(0, model_count):
-    #model_name = os.getenv("m{}_name".format(i))
-    #gdrive_id = os.getenv("m{}_griv".format(i))
-    #model_extname = os.getenv("m{}_type".format(i))
-    #model_desc = os.getenv("m{}_desc".format(i))
-
-    model_name = st.secrets["m{}_name".format(i)]
-    gdrive_id = st.secrets["m{}_griv".format(i)]
-    model_extname = st.secrets["m{}_type".format(i)]
-    model_desc = st.secrets["m{}_desc".format(i)]
+    if HuggingFace is False:
+        model_name = os.getenv("m{}_name".format(i))
+        gdrive_id = os.getenv("m{}_griv".format(i))
+        model_extname = os.getenv("m{}_type".format(i))
+        model_desc = os.getenv("m{}_desc".format(i))
+
+    else:
+        model_name = st.secrets["m{}_name".format(i)]
+        gdrive_id = st.secrets["m{}_griv".format(i)]
+        model_extname = st.secrets["m{}_type".format(i)]
+        model_desc = st.secrets["m{}_desc".format(i)]
+
     print(i, model_name, gdrive_id, model_extname, model_desc)
 
     path_model = os.path.join(base_download_path, model_name + model_extname)