Update app.py

app.py

@@ -93,9 +93,19 @@ def demo_retrieval():
     with tab_img:
         with st.form("rimgform"):
             k = st.slider("Number of items to retrieve", 1, 100, 16, key='rimage')
-            img = st.file_uploader("Upload an Image", key='rimageinput')
+            pic = st.file_uploader("Upload an Image", key='rimageinput')
             if st.form_submit_button("Retrieve with Image"):
                 prog.progress(0.49, "Computing Embeddings")
+                img = Image.open(pic)
+                st.image(img)
+                device = clip_model.device
+                tn = clip_prep(images=[img], return_tensors="pt").to(device)
+                enc = clip_model.get_image_features(pixel_values=tn['pixel_values'].type(half)).float().cpu()
+                
+                prog.progress(0.7, "Running Retrieval")
+                retrieval_results(retrieval.retrieve(enc, k, sim_th, filter_fn))
+                
+                prog.progress(1.0, "Idle")
 
     with tab_text:
         with st.form("rtextform"):