Update app.py

app.py

@@ -30,7 +30,7 @@ def generate_caption(protein, prompt):
     # os.system("python esm_scripts/extract.py esm2_t36_3B_UR50D /home/user/app/example.fasta /home/user/app --repr_layers 36 --truncation_seq_length 1024 --include per_tok")
     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)
+                       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")