Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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 |
|