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() | |