JayWadekar commited on
Commit
9437ceb
·
1 Parent(s): d995ec7
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,7 +4,7 @@
4
 
5
  from rag import RAG, load_docs
6
  from langchain_community.embeddings import HuggingFaceInstructEmbeddings
7
- from langchain.chat_models import ChatOpenAI
8
  import gradio as gr
9
 
10
  # Load the documentation
@@ -12,11 +12,12 @@ docs = load_docs()
12
  print("Pages loaded:", len(docs))
13
 
14
  # LLM model
15
- llm = ChatOpenAI(model="gpt-4o-mini")
16
 
17
  # Embeddings
18
  embed_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
19
  # embed_model = "nvidia/NV-Embed-v2"
 
20
  embeddings = HuggingFaceInstructEmbeddings(model_name=embed_model)
21
 
22
  # RAG chain
 
4
 
5
  from rag import RAG, load_docs
6
  from langchain_community.embeddings import HuggingFaceInstructEmbeddings
7
+ from langchain_community.chat_models import ChatOpenAI
8
  import gradio as gr
9
 
10
  # Load the documentation
 
12
  print("Pages loaded:", len(docs))
13
 
14
  # LLM model
15
+ llm = ChatOpenAI(model="gpt-4.1-mini-2025-04-14")
16
 
17
  # Embeddings
18
  embed_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
19
  # embed_model = "nvidia/NV-Embed-v2"
20
+ # text-embedding-3-small
21
  embeddings = HuggingFaceInstructEmbeddings(model_name=embed_model)
22
 
23
  # RAG chain