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