hf auth

app.py

@@ -144,7 +144,7 @@ def handle_name(name=None, pdb_input=None, model_version="ESM3"):
     return f'{pdb_name}-Dyna1{"" if model_version == "ESM3" else "-ESM2"}'
 
 @spaces.GPU(duration=300)
-def predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=False, model_version="ESM3", name=None):
+def predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=False, model_version="ESM3", name=None, oauth_token: Optional[str] = None):
     try:
         # Validate ESM2 requires sequence
         if model_version == "ESM2" and not sequence:
@@ -365,15 +365,21 @@ with dyna1_app:
         outputs=[esm3_inputs, esm2_inputs, results_esm3, results_esm2, examples_esm3]
     )
 
-    def predict_esm3(oauth_token: gr.OAuthToken | None, sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=False, model_version="ESM3", name=None):
-        check_permissions(oauth_token.token)
+    def predict_esm3(oauth_token: gr.OAuthToken | None, name, sequence, pdb_id, pdb_file, chain_id, use_pdb_seq):
+        if oauth_token is None:
+            raise gr.Error("Please log in to use this Space")
+
+        token_value = oauth_token.token
+        check_permissions(token_value)
+
         csv_output, pdb_output = predict_dynamics(
             sequence=sequence,
             pdb_input=pdb_id if pdb_id else pdb_file,
             chain_id=chain_id,
             use_pdb_seq=use_pdb_seq,
             model_version="ESM3",
-            name=name
+            name=name,
+            oauth_token=token_value
         )
         return [csv_output, pdb_output]
 
@@ -390,7 +396,7 @@ with dyna1_app:
 
     submit_btn_esm3.click(
         fn=predict_esm3,
-        inputs=[name_input, sequence_input_esm3, pdb_id, pdb_file, chain_id, use_pdb_seq],
+        inputs=[gr.OAuthToken(), name_input, sequence_input_esm3, pdb_id, pdb_file, chain_id, use_pdb_seq],
         outputs=[csv_output_esm3, pdb_output_esm3]
     )