Update app.py

app.py

@@ -23,6 +23,7 @@ model.to('cuda')
 def generate_caption(protein, prompt):
 
     esm_emb = torch.load('data/emb_esm2_3b/P18281.pt')['representations'][36]
+    torch.save(esm_emb, 'data/emb_esm2_3b/example.pt')
     '''
     inputs = tokenizer([protein], return_tensors="pt", padding=True, truncation=True).to('cuda')
     with torch.no_grad():