Update medical_rag.py

medical_rag.py

@@ -7,7 +7,7 @@ from transformers import (
     pipeline
 )
 from typing import List, Dict, Optional
-from general_rag import app, models, data, get_completion
+from general_rag import app, models, data, ChatQuery, get_completion
 
 # Initialize NLTK
 nltk.download('punkt')