Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,12 @@ from lavis.models.base_model import FAPMConfig
|
|
8 |
import spaces
|
9 |
import gradio as gr
|
10 |
from esm_scripts.extract import run_demo
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
# Load the model
|
13 |
model = Blip2ProteinMistral(config=FAPMConfig(), esm_size='3b')
|
@@ -22,7 +28,8 @@ def generate_caption(protein, prompt):
|
|
22 |
# f.write('>{}\n'.format("protein_name"))
|
23 |
# f.write('{}\n'.format(protein.strip()))
|
24 |
# 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")
|
25 |
-
esm_emb = run_demo(protein_name='protein_name', protein_seq=protein,
|
|
|
26 |
include='per_tok', repr_layers=36, truncation_seq_length=1024)
|
27 |
esm_emb = F.pad(esm_emb.t(), (0, 1024 - len(esm_emb))).t().to('cuda')
|
28 |
samples = {'name': ['protein_name'],
|
|
|
8 |
import spaces
|
9 |
import gradio as gr
|
10 |
from esm_scripts.extract import run_demo
|
11 |
+
from esm import pretrained
|
12 |
+
|
13 |
+
|
14 |
+
model_esm, alphabet = pretrained.load_model_and_alphabet('esm2_t36_3B_UR50D')
|
15 |
+
model_esm.eval()
|
16 |
+
model_esm = model.cuda()
|
17 |
|
18 |
# Load the model
|
19 |
model = Blip2ProteinMistral(config=FAPMConfig(), esm_size='3b')
|
|
|
28 |
# f.write('>{}\n'.format("protein_name"))
|
29 |
# f.write('{}\n'.format(protein.strip()))
|
30 |
# 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")
|
31 |
+
esm_emb = run_demo(protein_name='protein_name', protein_seq=protein,
|
32 |
+
model=model_esm, alphabet=alphabet,
|
33 |
include='per_tok', repr_layers=36, truncation_seq_length=1024)
|
34 |
esm_emb = F.pad(esm_emb.t(), (0, 1024 - len(esm_emb))).t().to('cuda')
|
35 |
samples = {'name': ['protein_name'],
|