Spaces:
Runtime error
Runtime error
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() |