NEXAS commited on
Commit
4373071
·
verified ·
1 Parent(s): 789107b

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +0 -2
src/streamlit_app.py CHANGED
@@ -71,8 +71,6 @@ if query_file is not None:
71
  query_img = Image.open(query_file)
72
  st.image(query_img, caption="Query Image", width=200)
73
  query_pre = st.session_state.preprocess(query_img).unsqueeze(0).to(st.session_state.device)
74
-
75
- to(st.session_state.device)
76
  with torch.no_grad():
77
  query_encoding = st.session_state.model.encode_image(query_pre)
78
 
 
71
  query_img = Image.open(query_file)
72
  st.image(query_img, caption="Query Image", width=200)
73
  query_pre = st.session_state.preprocess(query_img).unsqueeze(0).to(st.session_state.device)
 
 
74
  with torch.no_grad():
75
  query_encoding = st.session_state.model.encode_image(query_pre)
76