Update app.py

app.py

@@ -8,7 +8,6 @@ import os
 
 # Imports Hugging Face
 from huggingface_hub import hf_hub_download, login
-from google.colab import userdata
 
 # Imports locaux
 from modeling.BaseModel import BaseModel
@@ -22,7 +21,10 @@ from inference_utils.processing_utils import read_rgb, get_instances
 
 def init_huggingface():
     """Initialise la connexion Hugging Face et télécharge le modèle."""
-    login(userdata.get('HF_TOKEN'))
+    hf_token = os.getenv('HF_TOKEN')
+    if hf_token is None:
+        raise ValueError("Hugging Face token not found. Please set the HF_TOKEN environment variable.")
+    login(hf_token)
     return hf_hub_download(
         repo_id="microsoft/BiomedParse",
         filename="biomedparse_v1.pt",