File size: 1,106 Bytes
5bcc73a 67bc9b1 965645e 5bcc73a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import streamlit as st
st.set_page_config(
page_title="Grascii Search",
menu_items={
"About": """
Web interface for [grascii](https://github.com/grascii/grascii)'s
search utility
Image search powered by [gregg-vision-v0.2.1](https://huggingface.co/grascii/gregg-vision-v0.2.1)
"""
},
)
import pandas as pd # noqa E402
from search import write_grascii_search, write_reverse_search # noqa E402
pd.options.mode.copy_on_write = True
if "report_submitted" not in st.session_state:
st.session_state["report_submitted"] = False
if "grascii" not in st.session_state:
st.session_state["grascii"] = ""
if "alternatives" not in st.session_state:
st.session_state["alternatives"] = {}
if "save_image" in st.session_state:
st.session_state["save_image"] = st.session_state["save_image"]
if st.session_state["report_submitted"]:
st.toast("Thanks for the report!")
st.session_state["report_submitted"] = False
tab1, tab2 = st.tabs(["Grascii", "Reverse"])
with tab1:
write_grascii_search()
with tab2:
write_reverse_search()
|