rag-embedder / app.py
jackenmail's picture
Update app.py
07f4b32 verified
raw
history blame contribute delete
312 Bytes
# app.py (in your HF Space)
import gradio as gr
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("your-username/rag-embedder")
def embed(text):
return model.encode(text).tolist() # Returns vector
demo = gr.Interface(fn=embed, inputs="text", outputs="json")
demo.launch()