Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
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 |
|