pass HF token to structure features

app.py

@@ -22,7 +22,9 @@ from loguru import logger
 from matplotlib.cm import ScalarMappable
 from matplotlib.colors import Normalize
 
-from rocketshp import RocketSHP, load_sequence, load_structure
+from rocketshp import RocketSHP
+from rocketshp import load_sequence as get_sequence_features
+from rocketshp import load_structure as get_structure_features
 from rocketshp.network import (
     build_allosteric_network,
     calculate_centrality,
@@ -149,7 +151,9 @@ def predict_rocketshp(
             raise gr.Error("Sequence input is required for the selected model.")
 
         struct_features = None
-        seq_features = load_sequence(sequence, device=device, HF_TOKEN=token)
+
+        logger.info("Loading sequence features...")
+        seq_features = get_sequence_features(sequence, device=device, HF_TOKEN=token)
 
     else:
         if structure_file is None:
@@ -209,9 +213,12 @@ def predict_rocketshp(
         logger.info(len(structure))
         logger.info(structure[:3])
 
-        struct_features = load_structure(structure, device=device)
+        logger.info("Loading structure features...")
+        struct_features = get_structure_features(
+            structure, device=device, HF_TOKEN=token
+        )
         sequence = str(to_sequence(structure)[0][0])
-        seq_features = load_sequence(sequence, device=device)
+        seq_features = get_sequence_features(sequence, device=device, HF_TOKEN=token)
 
     # Make predictions
     with torch.no_grad():
@@ -429,18 +436,22 @@ def check_user_access(oauth_token: gr.OAuthToken | None):
         raise gr.Error("Please log in to use this Space")
 
     token = oauth_token.token
+    print(token)
 
     try:
         # Try to access a file from your private repo
         api = HfApi(token=token)
+        print(api)
 
         # Test access by trying to get repo info
-        api.repo_info(
+        repo_info = api.repo_info(
             repo_id="EvolutionaryScale/esm3-sm-open-v1",
             repo_type="model",  # or "dataset" or "space"
             token=token,
         )
 
+        print(repo_info)
+
         return (
             True,
             token,