Update app.py

app.py

@@ -51,7 +51,7 @@ class MyDataset(Dataset):
     def __init__(self,dict_data) -> None:
         super(MyDataset,self).__init__()
         self.data=dict_data
-        self.structure=pdb_structure(dict_data['test_sequences'])
+        self.structure=pdb_structure(dict_data['text'])
     def __getitem__(self, index):
         return self.data['text'][index], self.structure[index]
     def __len__(self):