Spaces:
Runtime error
Runtime error
attempting to add a sidebar
Browse files- pages/πββοΈ_Info.py +3 -0
- utils/frontend.py +7 -0
- utils/haystack.py +1 -0
- π‘_Home.py +2 -1
pages/πββοΈ_Info.py
CHANGED
|
@@ -1,4 +1,7 @@
|
|
| 1 |
import streamlit as st
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
st.markdown("""
|
| 4 |
# Better Image Retrieval With Reinforced CLIP π§
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
+
from utils.frontend import build_sidebar
|
| 3 |
+
|
| 4 |
+
build_sidebar()
|
| 5 |
|
| 6 |
st.markdown("""
|
| 7 |
# Better Image Retrieval With Reinforced CLIP π§
|
utils/frontend.py
CHANGED
|
@@ -1,5 +1,12 @@
|
|
| 1 |
import streamlit as st
|
| 2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
def set_state_if_absent(key, value):
|
| 4 |
if key not in st.session_state:
|
| 5 |
st.session_state[key] = value
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
|
| 3 |
+
def build_sidebar():
|
| 4 |
+
sidebar = """
|
| 5 |
+
<p><br/><a href='https://github.com/TuanaCelik/find-the-animal'>Github project</a> - Based on <a href='https://github.com/deepset-ai/haystack'>Haystack</a></p>
|
| 6 |
+
</div>
|
| 7 |
+
"""
|
| 8 |
+
st.sidebar.markdown(sidebar, unsafe_allow_html=True)
|
| 9 |
+
|
| 10 |
def set_state_if_absent(key, value):
|
| 11 |
if key not in st.session_state:
|
| 12 |
st.session_state[key] = value
|
utils/haystack.py
CHANGED
|
@@ -40,6 +40,7 @@ def start_haystack():
|
|
| 40 |
faiss_index_path=f"{INDEX_DIR}/images.faiss",
|
| 41 |
faiss_config_path=f"{INDEX_DIR}/images.json",
|
| 42 |
)
|
|
|
|
| 43 |
retriever_text = EmbeddingRetriever(
|
| 44 |
document_store=document_store_text,
|
| 45 |
embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
|
|
|
|
| 40 |
faiss_index_path=f"{INDEX_DIR}/images.faiss",
|
| 41 |
faiss_config_path=f"{INDEX_DIR}/images.json",
|
| 42 |
)
|
| 43 |
+
|
| 44 |
retriever_text = EmbeddingRetriever(
|
| 45 |
document_store=document_store_text,
|
| 46 |
embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
|
π‘_Home.py
CHANGED
|
@@ -6,7 +6,7 @@ from PIL import Image
|
|
| 6 |
from markdown import markdown
|
| 7 |
from annotated_text import annotation
|
| 8 |
from utils.haystack import query
|
| 9 |
-
from utils.frontend import reset_results, set_state_if_absent
|
| 10 |
|
| 11 |
def create_answer_objects(predictions):
|
| 12 |
results = []
|
|
@@ -32,6 +32,7 @@ def create_answer_objects(predictions):
|
|
| 32 |
return results
|
| 33 |
|
| 34 |
def main():
|
|
|
|
| 35 |
|
| 36 |
set_state_if_absent("statement", "What is the fastest animal?")
|
| 37 |
set_state_if_absent("results", None)
|
|
|
|
| 6 |
from markdown import markdown
|
| 7 |
from annotated_text import annotation
|
| 8 |
from utils.haystack import query
|
| 9 |
+
from utils.frontend import reset_results, set_state_if_absent, build_sidebar
|
| 10 |
|
| 11 |
def create_answer_objects(predictions):
|
| 12 |
results = []
|
|
|
|
| 32 |
return results
|
| 33 |
|
| 34 |
def main():
|
| 35 |
+
build_sidebar()
|
| 36 |
|
| 37 |
set_state_if_absent("statement", "What is the fastest animal?")
|
| 38 |
set_state_if_absent("results", None)
|