winfred2027 commited on
Commit
3779aed
·
verified ·
1 Parent(s): d39e5a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -86,6 +86,7 @@ def demo_retrieval():
86
  with st.form("rpcform"):
87
  k = st.slider("Number of items to retrieve", 1, 100, 16, key='rpc')
88
  #pc = utils.load_3D_shape('rpcinput')
 
89
  if st.form_submit_button("Retrieve with Point Cloud"):
90
  prog.progress(0.49, "Computing Embeddings")
91
 
@@ -94,6 +95,7 @@ def demo_retrieval():
94
  with st.form("rimgform"):
95
  k = st.slider("Number of items to retrieve", 1, 100, 16, key='rimage')
96
  pic = st.file_uploader("Upload an Image", key='rimageinput')
 
97
  if st.form_submit_button("Retrieve with Image"):
98
  prog.progress(0.49, "Computing Embeddings")
99
  img = Image.open(pic)
 
86
  with st.form("rpcform"):
87
  k = st.slider("Number of items to retrieve", 1, 100, 16, key='rpc')
88
  #pc = utils.load_3D_shape('rpcinput')
89
+ sim_th, filter_fn = retrieval_filter_expand('pc')
90
  if st.form_submit_button("Retrieve with Point Cloud"):
91
  prog.progress(0.49, "Computing Embeddings")
92
 
 
95
  with st.form("rimgform"):
96
  k = st.slider("Number of items to retrieve", 1, 100, 16, key='rimage')
97
  pic = st.file_uploader("Upload an Image", key='rimageinput')
98
+ sim_th, filter_fn = retrieval_filter_expand('image')
99
  if st.form_submit_button("Retrieve with Image"):
100
  prog.progress(0.49, "Computing Embeddings")
101
  img = Image.open(pic)