awinml's picture
Upload app.py
b84637f
raw
history blame
595 Bytes
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()