Update app.py

app.py

@@ -31,7 +31,9 @@ def generate_caption(protein, prompt):
     esm_emb = run_demo(protein_name='protein_name', protein_seq=protein, 
                        model=model_esm, alphabet=alphabet, 
                        include='per_tok', repr_layers=36, truncation_seq_length=1024)
+    print("esm embedding generated")
     esm_emb = F.pad(esm_emb.t(), (0, 1024 - len(esm_emb))).t().to('cuda')
+    print("esm embedding processed")
     samples = {'name': ['protein_name'],
                'image': torch.unsqueeze(esm_emb, dim=0),
                'text_input': ['none'],