oucgc1996 commited on
Commit
3a1dea8
1 Parent(s): b8f7dd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model_checkpoint = "facebook/esm2_t6_8M_UR50D"
14
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
15
 
16
  def AMP(file):
17
- test_sequences = file
18
  class MyDataset(Dataset):
19
  def __init__(self,dict_data) -> None:
20
  super(MyDataset,self).__init__()
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
15
 
16
  def AMP(file):
17
+ test_sequences = [file]
18
  class MyDataset(Dataset):
19
  def __init__(self,dict_data) -> None:
20
  super(MyDataset,self).__init__()