File size: 473 Bytes
e468f5a a46f28b e468f5a a46f28b e468f5a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# -*- 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() |