gelnesr commited on
Commit
779c798
·
1 Parent(s): 4f3ef07
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -368,7 +368,11 @@ with dyna1_app:
368
  )
369
 
370
  def predict_esm3(oauth_token: gr.OAuthToken | None, name, sequence, pdb_id, pdb_file, chain_id, use_pdb_seq):
371
- check_permissions(oauth_token.token)
 
 
 
 
372
  csv_output, pdb_output = predict_dynamics(
373
  sequence=sequence,
374
  pdb_input=pdb_id if pdb_id else pdb_file,
@@ -392,7 +396,7 @@ with dyna1_app:
392
 
393
  submit_btn_esm3.click(
394
  fn=predict_esm3,
395
- inputs=[gr.OAuthToken, name_input, sequence_input_esm3, pdb_id, pdb_file, chain_id, use_pdb_seq],
396
  outputs=[csv_output_esm3, pdb_output_esm3]
397
  )
398
 
 
368
  )
369
 
370
  def predict_esm3(oauth_token: gr.OAuthToken | None, name, sequence, pdb_id, pdb_file, chain_id, use_pdb_seq):
371
+ if gr.OAuthToken not None:
372
+ check_permissions(gr.OAuthToken.token)
373
+ else:
374
+ raise gr.Error("Please log in to use this Space")
375
+
376
  csv_output, pdb_output = predict_dynamics(
377
  sequence=sequence,
378
  pdb_input=pdb_id if pdb_id else pdb_file,
 
396
 
397
  submit_btn_esm3.click(
398
  fn=predict_esm3,
399
+ inputs=[name_input, sequence_input_esm3, pdb_id, pdb_file, chain_id, use_pdb_seq],
400
  outputs=[csv_output_esm3, pdb_output_esm3]
401
  )
402