Towhidul commited on
Commit
92e88b1
·
verified ·
1 Parent(s): 3948cf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -29,7 +29,7 @@ from llama_index.core.schema import ImageNode
29
  from langchain_core.prompts import ChatPromptTemplate
30
  from langchain_core.prompts import HumanMessagePromptTemplate
31
  from langchain_core.messages import SystemMessage
32
-
33
  nest_asyncio.apply()
34
 
35
  # Setting API keys
@@ -123,8 +123,8 @@ else:
123
  text_nodes = []
124
 
125
  # Setup index and LLM
126
- embed_model = OpenAIEmbedding(model="text-embedding-3-large")
127
- llm = OpenAI("gpt-4o-mini-2024-07-18")
128
  Settings.llm = llm
129
  Settings.embed_model = embed_model
130
 
 
29
  from langchain_core.prompts import ChatPromptTemplate
30
  from langchain_core.prompts import HumanMessagePromptTemplate
31
  from langchain_core.messages import SystemMessage
32
+ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
33
  nest_asyncio.apply()
34
 
35
  # Setting API keys
 
123
  text_nodes = []
124
 
125
  # Setup index and LLM
126
+ embed_model = HuggingFaceEmbedding(model="neuml/pubmedbert-base-embeddings")
127
+ llm = OpenAI(model="gpt-4o-mini-2024-07-18", temperature=0.1)
128
  Settings.llm = llm
129
  Settings.embed_model = embed_model
130