Tamil Eniyan commited on
Commit
114c773
·
1 Parent(s): 20c970d

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-base-squad2" # You can change this to any Hugging Face QA model
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():