Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import faiss
|
|
6 |
|
7 |
|
8 |
if "initialized" not in st.session_state:
|
9 |
-
st.session_state.dataset = datasets.load_dataset('
|
10 |
st.session_state.all_authors = list(set(st.session_state.dataset['author']))
|
11 |
model_name = "BAAI/bge-small-en-v1.5" # "Cohere/Cohere-embed-english-light-v3.0" # "sentence-transformers/all-MiniLM-L6-v2"
|
12 |
st.session_state.encoder = SentenceTransformer(model_name)
|
|
|
6 |
|
7 |
|
8 |
if "initialized" not in st.session_state:
|
9 |
+
st.session_state.dataset = datasets.load_dataset('m-ric/english_historical_quotes', download_mode="force_redownload")['train']
|
10 |
st.session_state.all_authors = list(set(st.session_state.dataset['author']))
|
11 |
model_name = "BAAI/bge-small-en-v1.5" # "Cohere/Cohere-embed-english-light-v3.0" # "sentence-transformers/all-MiniLM-L6-v2"
|
12 |
st.session_state.encoder = SentenceTransformer(model_name)
|