Update app.py

app.py

@@ -3,13 +3,12 @@ from transformers import AutoTokenizer, EsmModel
 import torch
 import json
 
-
+# selecing and loading a model
 model_name = st.selectbox(
     'Choose a model',
     ["facebook/esm2_t6_8M_UR50D", "facebook/esm2_t48_15B_UR50D"])
 
 st.write('You selected model:', model_name)
-
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = EsmModel.from_pretrained(model_name)
 
@@ -29,16 +28,20 @@ def embed(aa_seq):
     }
     json_data = json.dumps(data)
 
-
-st.download_button(
-    label="Download JSON file",
-    data=json_data,
-    file_name="esm-2 last hidden states.json",
-    mime='application/json'
-)
+    st.download_button(
+        label="Download JSON file",
+        data=json_data,
+        file_name="esm-2 last hidden states.json",
+        mime='application/json'
+    )
 
     
-if st.button('Run Function'):
+if st.button('Get embedding'):
     embed(aa_seq)
 
 st.write('Also, Dania is not gay')
+#uploading AA sequences file
+uploaded_file = st.file_uploader("Choose a JSON file", type='json')
+if uploaded_file is not None:
+    data = json.load(uploaded_file)
+    st.write(data)