Hugging Face's logo

Spaces:
savakholin
/
esm-2
Paused

App Files Community
savakholin commited on
Commit
6d84961
·
1 Parent(s): bf6dd26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,10 +3,9 @@ from transformers import AutoTokenizer, EsmModel
3
  import torch
4
 
5
 
6
- model_name = st.multiselect(
7
  'Choose a model',
8
- ["facebook/esm2_t6_8M_UR50D", "facebook/esm2_t48_15B_UR50D"],
9
- ["facebook/esm2_t6_8M_UR50D"])
10
 
11
  st.write('You selected model:', model_name)
12
 
 
3
  import torch
4
 
5
 
6
+ model_name = st.selectbox(
7
  'Choose a model',
8
+ ["facebook/esm2_t6_8M_UR50D", "facebook/esm2_t48_15B_UR50D"])
 
9
 
10
  st.write('You selected model:', model_name)
11