9 lines
213 B
Python
9 lines
213 B
Python
from llama_index.core import Document, VectorStoreIndex
|
|
|
|
import mlflow
|
|
|
|
index = VectorStoreIndex.from_documents(documents=[Document.example()])
|
|
retriever = index.as_retriever()
|
|
|
|
mlflow.models.set_model(retriever)
|