realrastayouth's picture
Update app.py
aec45ab
raw
history blame
No virus
693 Bytes
import gradio as gr
import pandas as pd
import faiss
from sentence_transformers import SentenceTransformer
#load the sentences data
sentences_df = pd.read_pickle("./sentences_data.pkl")
#load the index
index = faiss.read_index("semantic_search_vector.index")
def question_answer(query):
model = SentenceTransformer('bert-base-nli-mean-tokens')
k = 4
xq = model.encode([query])
D, I = index.search(xq, k) # search
sentences_df["sentences"].iloc[I.tolist()[0]]
a = list(sentences_df["sentences"].iloc[I.tolist()[0]])
results = "\n".join(a)
answer = results
return answer
iface = gr.Interface(fn=question_answer, inputs="text", outputs="text").launch()