cb1716pics commited on
Commit
147795a
·
verified ·
1 Parent(s): 3c7eb9f

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +0 -3
  2. retrieval.py +5 -1
app.py CHANGED
@@ -5,9 +5,6 @@ from evaluation import calculate_metrics
5
  from data_processing import load_recent_questions, save_recent_question
6
  import time
7
  import matplotlib.pyplot as plt
8
- import nltk
9
-
10
- nltk.download('punkt')
11
 
12
  # Page Title
13
  st.title("RAG7 - Real World RAG System")
 
5
  from data_processing import load_recent_questions, save_recent_question
6
  import time
7
  import matplotlib.pyplot as plt
 
 
 
8
 
9
  # Page Title
10
  st.title("RAG7 - Real World RAG System")
retrieval.py CHANGED
@@ -5,10 +5,14 @@ import faiss
5
  from rank_bm25 import BM25Okapi
6
  from data_processing import embedding_model
7
  from sentence_transformers import CrossEncoder
8
- from nltk.tokenize import word_tokenize
9
  import string
10
  import nltk
11
 
 
 
 
 
 
12
  reranker = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
13
 
14
  retrieved_docs = None
 
5
  from rank_bm25 import BM25Okapi
6
  from data_processing import embedding_model
7
  from sentence_transformers import CrossEncoder
 
8
  import string
9
  import nltk
10
 
11
+ import nltk
12
+ nltk.download('punkt')
13
+
14
+ from nltk.tokenize import word_tokenize
15
+
16
  reranker = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
17
 
18
  retrieved_docs = None