|
import streamlit as st |
|
from transformers import AutoTokenizer, EsmModel |
|
import torch |
|
|
|
model_name = "facebook/esm2_t6_8M_UR50D" |
|
tokenizer = AutoTokenizer.from_pretrained(model_name) |
|
model = EsmModel.from_pretrained(model_name) |
|
|
|
aa_seq = st.text_input('Type AA sequance here') |
|
|
|
inputs = tokenizer(aa_seq, return_tensors="pt") |
|
outputs = model(**inputs) |
|
|
|
last_hidden_states = outputs.last_hidden_state |
|
|
|
st.write(model_name, 'last hidden states',last_hidden_states.shape, last_hidden_states, 'Also, Dania is not gay') |
|
|