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 |
|