Update app.py

app.py

@@ -1,4 +1,16 @@
 import streamlit as st
+from transformers import AutoTokenizer, EsmModel
+import torch
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x, '. Also, Dania is gay')
+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')