1MB1 commited on
Commit
59eeb97
·
verified ·
1 Parent(s): 0a6a3b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,10 +100,10 @@ async def upload_file(file: UploadFile = File(...)):
100
  nodes = parser.get_nodes_from_documents(documents)
101
 
102
  # Create or update vector index
103
- Settings.embed_model = HuggingFaceEmbedding(model_name="WhereIsAI/UAE-Large-V1")
104
  # if vector_index is None:
105
  if not hasattr(app.state, "vector_index") or app.state.vector_index is None:
106
- app.state.vector_index = VectorStoreIndex(nodes, embed_model=Settings.embed_model)
107
  message = "New vector index created and file stored."
108
  else:
109
  app.state.vector_index.insert_nodes(nodes)
 
100
  nodes = parser.get_nodes_from_documents(documents)
101
 
102
  # Create or update vector index
103
+ embed_model = HuggingFaceEmbedding(model_name="WhereIsAI/UAE-Large-V1")
104
  # if vector_index is None:
105
  if not hasattr(app.state, "vector_index") or app.state.vector_index is None:
106
+ app.state.vector_index = VectorStoreIndex(nodes, embed_model=embed_model)
107
  message = "New vector index created and file stored."
108
  else:
109
  app.state.vector_index.insert_nodes(nodes)