Spaces:
Running
Running
JayWadekar
commited on
Commit
·
9437ceb
1
Parent(s):
d995ec7
gpt 4.1
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@
|
|
4 |
|
5 |
from rag import RAG, load_docs
|
6 |
from langchain_community.embeddings import HuggingFaceInstructEmbeddings
|
7 |
-
from
|
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-
|
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
|