barabum's picture
init
1f9d7aa
raw
history blame
871 Bytes
import gradio as gr
import openai
import numpy as np
def calc_distance(text1, text2, api) -> str:
text1_emb = openai.Embedding.create(input=text1, model="text-embedding-ada-002", api_key=api)
text2_emb = openai.Embedding.create(input=text2, model="text-embedding-ada-002", api_key=api)
text1_emb = np.array(text1_emb)
text2_emb = np.array(text2_emb)
distance = np.linalg.norm(text1_emb-text2_emb)
return str(distance)
with gr.Blocks() as b:
openai_apikey = gr.TextArea(label="OpenAI API key", lines=1)
with gr.Row():
with gr.Column():
text1 = gr.TextArea(label="Text 1")
text2 = gr.TextArea(label="Text 2")
btn = gr.Button("Calculate")
output = gr.outputs.Label(label="Distance")
btn.click(fn=calc_distance, inputs=(text1, text2, openai_apikey), outputs=output)
b.launch()