Configure for HF Spaces

Signed-off-by: Snehil Shah <snehilshah.989@gmail.com>
Snehil-Shah · Jan 8, 2024 · 77d7e87 · 77d7e87
1 parent ba6c86a
commit 77d7e87
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -1,13 +1,14 @@
 import gradio as gr
+import os
 from qdrant_client import QdrantClient
 from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer("clip-ViT-B-32")
 
 qdrant_client = QdrantClient(
-    url = 'url',
+    url = os.environ['QDRANT_URL'],
     port= 443,
-    api_key = "key",
+    api_key = os.environ['QDRANT_API_KEY']
 )
 
 def search_images(modality, count, input_text, input_image):