Update app.py

app.py

@@ -18,10 +18,13 @@ model.to('cuda')
 @spaces.GPU
 def generate_caption(protein, prompt):
     # Process the image and the prompt
+    print(f"system path: {os.getcwd()}")
     with open('data/fasta/example.fasta', 'w') as f:
         f.write('>{}\n'.format("protein_name"))
         f.write('{}\n'.format(protein.strip()))
+    print(f"fasta prepared")
     os.system("python esm_scripts/extract.py esm2_t36_3B_UR50D data/fasta/example.fasta data/emb_esm2_3b --repr_layers 36 --truncation_seq_length 1024 --include per_tok")
+    print(f"protein pt file prepared")
     esm_emb = torch.load("data/emb_esm2_3b/protein_name.pt")['representations'][36]
     esm_emb = F.pad(esm_emb.t(), (0, 1024 - len(esm_emb))).t().to('cuda')
     samples = {'name': ['test_protein'],