yurapodk commited on
Commit
a79b50e
·
1 Parent(s): dfd5c5e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ if(input_image):
18
  st.image(input_image, use_column_width=True)
19
 
20
  detection_model = st.selectbox("Select detection model", ("YOLO", ))
21
- embedding_model = st.selectbox("Select embedding model", ("OML", ))
22
 
23
  if st.button("Get prediction"):
24
  with st.spinner("Loading..."):
 
18
  st.image(input_image, use_column_width=True)
19
 
20
  detection_model = st.selectbox("Select detection model", ("YOLO", ))
21
+ embedding_model = st.selectbox("Select embedding model", ("vits8", ))
22
 
23
  if st.button("Get prediction"):
24
  with st.spinner("Loading..."):