JayWadekar commited on
Commit
7b527d5
·
1 Parent(s): 041aa44
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,9 +3,9 @@
3
  # using Langchain and deployed with Gradio
4
 
5
  from rag import RAG, load_docs
6
- from langchain_google_genai import ChatGoogleGenerativeAI
7
  from langchain_community.embeddings import HuggingFaceInstructEmbeddings
8
- #from langchain.chat_models import ChatOpenAI
9
  import gradio as gr
10
  import os
11
 
@@ -14,8 +14,8 @@ docs = load_docs()
14
  print("Pages loaded:", len(docs))
15
 
16
  # LLM model
17
- #llm = ChatOpenAI(model="gpt-4o-mini")
18
- llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
19
 
20
  # Embeddings
21
  embed_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
 
3
  # using Langchain and deployed with Gradio
4
 
5
  from rag import RAG, load_docs
6
+ # from langchain_google_genai import ChatGoogleGenerativeAI
7
  from langchain_community.embeddings import HuggingFaceInstructEmbeddings
8
+ from langchain.chat_models import ChatOpenAI
9
  import gradio as gr
10
  import os
11
 
 
14
  print("Pages loaded:", len(docs))
15
 
16
  # LLM model
17
+ llm = ChatOpenAI(model="gpt-4o-mini")
18
+ # llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
19
 
20
  # Embeddings
21
  embed_model = "sentence-transformers/multi-qa-distilbert-cos-v1"