Update app.py

app.py

@@ -37,9 +37,11 @@ def generate_caption(protein, prompt):
     repr_layers=[36]
     truncation_seq_length=1024
     toks_per_batch=4096
-    
+    print("start")
     dataset = FastaBatchedDataset([protein_name], [protein_seq])
+    print("dataset prepared")
     batches = dataset.get_batch_indices(toks_per_batch, extra_toks_per_seq=1)
+    print("batches prepared")
     data_loader = torch.utils.data.DataLoader(
         dataset, collate_fn=alphabet.get_batch_converter(truncation_seq_length), batch_sampler=batches
     )