import gradio as gr
import numpy as np
from InstructorEmbedding import INSTRUCTOR

model = INSTRUCTOR('hkunlp/instructor-xl')

def create_embedding(query_instruction, query):
    embeddings = model.encode([[query_instruction, query]])
    embeddings_array = np.array(embeddings)
    return embeddings_array

instructor_model_embeddings = gr.Interface(
    fn=create_embedding,
    inputs=[
        gr.inputs.Textbox(label="Query_Instruction"),
        gr.inputs.Textbox(label="Query")
    ],
    outputs=gr.Dataframe(type="numpy", datatype="number"),
    title="API-Instructor-XL-1",
).launch()