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['did'][i]
    return name
    
gr.Interface(
    predict,
    inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
    outputs=gr.outputs.Label(num_top_classes=2),
    title="Hot Dog? Or Not?",
).launch()

gr.Interface(fn = predict, inputs = [gr.Textbox(label="Description:", placeholder="z.B. Schaltbefehl Pumpe", lines=1), lines=1)], 
             title = 'ECLASS-Suche',
             outputs = gr.outputs.Label(name)).launch()