Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files- app.py +0 -3
- 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
|