Spaces:

stamps-labs
/

swp-ui

Runtime error

yurapodk commited on Jul 2, 2023

Commit

3ed339a

1 Parent(s): 2869667

a

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ if(input_image):
     st.header("Original")
     st.image(input_image, use_column_width=True)
-    detection_model = st.selectbox("Select detection model", ("YOLO", ))
     embedding_model = st.selectbox("Select embedding model", ("vits8", ))
     if st.button("Get prediction"):

     st.header("Original")
     st.image(input_image, use_column_width=True)
+    detection_model = st.selectbox("Select detection model", ("YOLO-stamp", ))
     embedding_model = st.selectbox("Select embedding model", ("vits8", ))
     if st.button("Get prediction"):