hf auth

app.py

@@ -27,7 +27,6 @@ REPO_URL = "https://github.com/WaymentSteeleLab/Dyna-1.git"
 DYNA_MODEL_ID = "gelnesr/Dyna-1"
 
 def setup_environment():
-    """Setup the environment by cloning the repo and downloading model weights."""
     base_dir = Path(os.getcwd())
     dyna1_dir = base_dir / "Dyna-1"
 
@@ -44,9 +43,6 @@ def setup_environment():
                 print(f"Successfully downloaded {filename}")
             except Exception as e:
                 print(f"Error downloading {filename}: {str(e)}")
-                if not os.getenv("SPACE_ID"):
-                    print("Please ensure you have access to the model weights and are logged in to HuggingFace.")
-                    print("You can login using `huggingface-cli login` or by setting the HUGGING_FACE_HUB_TOKEN environment variable.")
 
     return dyna1_dir
 
@@ -365,10 +361,9 @@ with dyna1_app:
         outputs=[esm3_inputs, esm2_inputs, results_esm3, results_esm2, examples_esm3]
     )
 
-    def predict_esm3(oauth_token: gr.OAuthToken | None, name, sequence, pdb_id, pdb_file, chain_id, use_pdb_seq):
+    def predict_esm3(name, sequence, pdb_id, pdb_file, chain_id, use_pdb_seq, oauth_token: gr.OAuthToken | None,=None):
         if oauth_token is None:
             raise gr.Error("Please log in to use this Space")
-
         token_value = oauth_token.token
         check_permissions(token_value)
 
@@ -396,7 +391,7 @@ with dyna1_app:
 
     submit_btn_esm3.click(
         fn=predict_esm3,
-        inputs=[gr.OAuthToken(), name_input, sequence_input_esm3, pdb_id, pdb_file, chain_id, use_pdb_seq],
+        inputs=[name_input, sequence_input_esm3, pdb_id, pdb_file, chain_id, use_pdb_seq],
         outputs=[csv_output_esm3, pdb_output_esm3]
     )