| # -*- coding: utf-8 -*- | |
| """ | |
| Created on Wed Mar 22 19:59:54 2023 | |
| """ | |
| import gradio as gr | |
| from simiandb import Simiandb | |
| from langchain.embeddings import HuggingFaceEmbeddings | |
| model_name = "all-MiniLM-L6-v2" | |
| hf = HuggingFaceEmbeddings(model_name=model_name) | |
| documentdb = Simiandb("mystore", embedding_function=hf, mode="a") | |
| def search(query): | |
| return documentdb.similarity_search(query) | |
| iface = gr.Interface(fn=search, inputs="text", outputs="text") | |
| iface.launch() |