import gradio as gr
import sentence_transformers
from sentence_transformers import SentenceTransformer

import pandas as pd

model = SentenceTransformer('JoBeer/all-mpnet-base-v2-eclass')

corpus = pd.read_excel('corpus.xlsx')

def predict(i):
    name = corpus[i][5]
    return name

interface = gr.Interface(fn = predict, inputs = gr.Textbox(label="did:", placeholder="z.B. query", lines=1), 
            outputs = 'text',
            title = 'ECLASS-Property-Search')
    
interface.launch()