awinml's picture
Upload 2 files
28e2a5b
raw
history blame
518 Bytes
import gradio as gr
from InstructorEmbedding import INSTRUCTOR
model = INSTRUCTOR('hkunlp/instructor-xl')
def create_embedding(instruction, sentence):
embeddings = model.encode([[instruction, sentence]])
print("Embeddings:", embeddings)
return embeddings
instructor_model_embeddings = gr.Interface(
fn=create_embedding,
inputs=[
gr.inputs.Textbox(label="Query_Instruction"),
gr.inputs.Textbox(label="Query")
],
outputs="text",
title="API-Instructor-XL-1",
).launch()