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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -103,7 +103,7 @@ async def upload_file(file: UploadFile = File(...)):
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=embed_model)
107
  message = "New vector index created and file stored."
108
  else:
109
  app.state.vector_index.insert_nodes(nodes)
 
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)