Spaces:
Sleeping
Sleeping
Update medical_rag.py
Browse files- medical_rag.py +1 -1
medical_rag.py
CHANGED
@@ -7,7 +7,7 @@ from transformers import (
|
|
7 |
pipeline
|
8 |
)
|
9 |
from typing import List, Dict, Optional
|
10 |
-
from general_rag import app, models, data, ChatQuery, get_completion
|
11 |
|
12 |
# Initialize NLTK
|
13 |
nltk.download('punkt')
|
|
|
7 |
pipeline
|
8 |
)
|
9 |
from typing import List, Dict, Optional
|
10 |
+
from general_rag import app, models, data, ChatQuery, embed_query_text, get_completion, load_embeddings, translate_text, retrieve_document_texts, query_embeddings
|
11 |
|
12 |
# Initialize NLTK
|
13 |
nltk.download('punkt')
|