Update app.py

app.py

@@ -10,6 +10,8 @@ import base64
 import os
 import huggingface_hub
 from huggingface_hub import hf_hub_download
+from huggingface_hub import login
+
 
 
 # Load label mapping
@@ -52,14 +54,20 @@ def load_model():
 
     #model_weights_path = os.getenv('MODEL_PATH')
     #weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
-
-
-    # Download the file
-    file_path = hf_hub_download(repo_id="mawairon/noo_test", filename="/gena-blastln-bs33-lr4e-05-S168.pth")
-
-    # Now you can use the file_path to load your .pth file
-    weights = torch.load(file_path)
-
+    
+
+    # Ensure you are authenticated if the repository is private
+    token = os.getenv('HUGGINGFACE_TOKEN')
+    if token is None:
+        raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
+
+    login(token=token)
+    file_path = hf_hub_download(
+        repo_id="mawairon/noo_test",
+        filename="gena-blastln-bs33-lr4e-05-S168.pth",
+        use_auth_token=token
+    )
+    weights = torch.load(file_path, map_location=torch.device('cpu'))
 
     base_model.load_state_dict(weights['model_state_dict'])
     log_reg.load_state_dict(weights['log_reg_state_dict'])