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