Update src/streamlit_app.py

src/streamlit_app.py

@@ -70,7 +70,7 @@ query_file = st.file_uploader("Choose a query image", type=['png', 'jpg', 'jpeg'
 if query_file is not None:
     query_img = Image.open(query_file)
     st.image(query_img, caption="Query Image", width=200)
-    query_pre = st.session_state.preprocess(query_img).unsqueeze(0).
+    query_pre = st.session_state.preprocess(query_img).unsqueeze(0).to(st.session_state.device)
 
 to(st.session_state.device)
     with torch.no_grad():