Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ model.to('cuda')
|
|
23 |
def generate_caption(protein, prompt):
|
24 |
|
25 |
esm_emb = torch.load('data/emb_esm2_3b/P18281.pt')['representations'][36]
|
|
|
26 |
'''
|
27 |
inputs = tokenizer([protein], return_tensors="pt", padding=True, truncation=True).to('cuda')
|
28 |
with torch.no_grad():
|
|
|
23 |
def generate_caption(protein, prompt):
|
24 |
|
25 |
esm_emb = torch.load('data/emb_esm2_3b/P18281.pt')['representations'][36]
|
26 |
+
torch.save(esm_emb, 'data/emb_esm2_3b/example.pt')
|
27 |
'''
|
28 |
inputs = tokenizer([protein], return_tensors="pt", padding=True, truncation=True).to('cuda')
|
29 |
with torch.no_grad():
|