thechaiexperiment commited on
Commit
c21e3a0
·
verified ·
1 Parent(s): d3ee370

Update medical_rag.py

Browse files
Files changed (1) hide show
  1. 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')