Update app.py

app.py

@@ -2,8 +2,15 @@ import gradio as gr
 from huggingface_hub import get_token, InferenceClient
 from sentence_transformers import CrossEncoder
 
-inference_client = InferenceClient(api_key=get_token())
+
+gradio_client = Client("https://smol-blueprint-vector-search-hub.hf.space/")
 reranker = CrossEncoder("sentence-transformers/all-MiniLM-L12-v2")
+inference_client = InferenceClient(api_key=get_token())
+
+
+def similarity_search(query: str, k: int = 5):
+    results = gradio_client.predict(api_name="/similarity_search", query=query, k=k)
+    return pd.DataFrame(data=results["data"], columns=results["headers"])
 
 
 def query_and_rerank_documents(query: str, k_retrieved: int = 10):