mammal collage for Field Museum

app.py

@@ -271,8 +271,8 @@ def handle_image_upload_and_gallery_hf(uploaded_files):
                 return True
             
             # Print out details of the uploaded file for debugging
-            st.write(f"Uploaded file: {uploaded_file.name}")
-            st.write(f"File size: {len(uploaded_file.getvalue())} bytes")
+            # st.write(f"Uploaded file: {uploaded_file.name}")
+            # st.write(f"File size: {len(uploaded_file.getvalue())} bytes")
 
             # Check if the uploaded file is not empty
             if len(uploaded_file.getvalue()) == 0:

vouchervision/API_validation.py

@@ -318,7 +318,6 @@ class APIvalidation:
             k_OPENAI_API_KEY = self.cfg_private['openai']['OPENAI_API_KEY']
             k_openai_azure = self.cfg_private['openai_azure']['OPENAI_API_KEY_AZURE']
 
-            k_huggingface = self.cfg_private['huggingface']['hf_token']
 
             k_project_id = self.cfg_private['google']['GOOGLE_PROJECT_ID']
             k_location = self.cfg_private['google']['GOOGLE_LOCATION']
@@ -328,7 +327,9 @@ class APIvalidation:
             k_here = self.cfg_private['here']['API_KEY']
             k_opencage = self.cfg_private['open_cage_geocode']['API_KEY']
 
-            
+            k_huggingface = self.cfg_private['huggingface']['hf_token']
+            os.environ["HUGGING_FACE_KEY"] = k_huggingface
+
 
 
         # Check each key and add to the respective list

vouchervision/LLM_local_MistralAI.py

@@ -36,7 +36,12 @@ class LocalMistralHandler:
 
         self.model_name = model_name
         self.model_id = f"mistralai/{self.model_name}"
-        self.model_path = hf_hub_download(repo_id=self.model_id, repo_type="model", filename="config.json", use_auth_token=os.getenv("HUGGING_FACE_KEY"))
+        huggingface_token = os.getenv("HUGGING_FACE_KEY")
+        if not huggingface_token:
+            self.logger.error("Hugging Face token is not set. Please set it using the HUGGING_FACE_KEY environment variable.")
+            raise ValueError("Hugging Face token is not set.")
+        
+        self.model_path = hf_hub_download(repo_id=self.model_id, repo_type="model", filename="config.json", token=huggingface_token)
 
         self.JSON_dict_structure = JSON_dict_structure
         self.starting_temp = float(self.STARTING_TEMP)

vouchervision/LLM_local_cpu_MistralAI.py

@@ -48,12 +48,18 @@ class LocalCPUMistralHandler:
         self.model_name = "Mistral-7B-Instruct-v0.2-GGUF"  #huggingface-cli download TheBloke/Mistral-7B-Instruct-v0.2-GGUF mistral-7b-instruct-v0.2.Q4_K_M.gguf --local-dir /home/brlab/.cache --local-dir-use-symlinks False
         self.model_id = f"TheBloke/{self.model_name}"
         name_parts = self.model_name.split('-')
+
+        huggingface_token = os.getenv("HUGGING_FACE_KEY")
+        if not huggingface_token:
+            self.logger.error("Hugging Face token is not set. Please set it using the HUGGING_FACE_KEY environment variable.")
+            raise ValueError("Hugging Face token is not set.")
         
         if self.model_name == "Mistral-7B-Instruct-v0.2-GGUF":
             self.model_file = 'mistral-7b-instruct-v0.2.Q4_K_M.gguf'
             self.model_path = hf_hub_download(repo_id=self.model_id,
                                  filename=self.model_file,
-                                 repo_type="model")
+                                 repo_type="model",
+                                 token=huggingface_token)
         else:
             raise f"Unsupported GGUF model name"
 

vouchervision/utils_VoucherVision.py

@@ -472,6 +472,7 @@ class VoucherVision():
             k_openai_azure = self.cfg_private['openai_azure']['OPENAI_API_KEY_AZURE']
 
             k_huggingface = self.cfg_private['huggingface']['hf_token']
+            os.environ["HUGGING_FACE_KEY"] = k_huggingface
 
             k_google_project_id = self.cfg_private['google']['GOOGLE_PROJECT_ID']
             k_google_location = self.cfg_private['google']['GOOGLE_LOCATION']