Tamil Eniyan
commited on
Commit
·
114c773
1
Parent(s):
20c970d
Add application file
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from transformers import pipeline
|
|
9 |
INDEX_FILE = "faiss_index.index"
|
10 |
CHUNKS_FILE = "chunks.pkl"
|
11 |
EMBEDDING_MODEL_NAME = "all-MiniLM-L6-v2"
|
12 |
-
QA_MODEL_NAME = "deepset/roberta-
|
13 |
|
14 |
@st.cache_resource
|
15 |
def load_index_and_chunks():
|
|
|
9 |
INDEX_FILE = "faiss_index.index"
|
10 |
CHUNKS_FILE = "chunks.pkl"
|
11 |
EMBEDDING_MODEL_NAME = "all-MiniLM-L6-v2"
|
12 |
+
QA_MODEL_NAME = "deepset/roberta-large-squad2" # You can change this to any Hugging Face QA model
|
13 |
|
14 |
@st.cache_resource
|
15 |
def load_index_and_chunks():
|