Spaces:
Sleeping
Sleeping
adds timeout
Browse files- app.py +2 -2
- neural_searcher.py +1 -1
app.py
CHANGED
|
@@ -16,8 +16,8 @@ neural_searcher = NeuralSearcher(collection_name=os.getenv('COLLECTION_NAME'))
|
|
| 16 |
REQUEST_TIMEOUT_ERROR = 1
|
| 17 |
|
| 18 |
@app.get("/api/search")
|
| 19 |
-
|
| 20 |
-
data =
|
| 21 |
return data
|
| 22 |
|
| 23 |
|
|
|
|
| 16 |
REQUEST_TIMEOUT_ERROR = 1
|
| 17 |
|
| 18 |
@app.get("/api/search")
|
| 19 |
+
def search(q: str):
|
| 20 |
+
data = neural_searcher.search(text=q)
|
| 21 |
return data
|
| 22 |
|
| 23 |
|
neural_searcher.py
CHANGED
|
@@ -13,7 +13,7 @@ class NeuralSearcher:
|
|
| 13 |
self.late_interaction_model = LateInteractionTextEmbedding(os.getenv('LATE_INTERACTION_MODEL'))
|
| 14 |
self.qdrant_client = QdrantClient(os.getenv('QDRANT_URL'),api_key=os.getenv('QDRANT_API_KEY'),https=True)
|
| 15 |
|
| 16 |
-
|
| 17 |
|
| 18 |
dense_query = self.dense_model.encode(text).tolist()
|
| 19 |
sparse_query = next(self.sparse_model.query_embed(text))
|
|
|
|
| 13 |
self.late_interaction_model = LateInteractionTextEmbedding(os.getenv('LATE_INTERACTION_MODEL'))
|
| 14 |
self.qdrant_client = QdrantClient(os.getenv('QDRANT_URL'),api_key=os.getenv('QDRANT_API_KEY'),https=True)
|
| 15 |
|
| 16 |
+
def search(self, text: str):
|
| 17 |
|
| 18 |
dense_query = self.dense_model.encode(text).tolist()
|
| 19 |
sparse_query = next(self.sparse_model.query_embed(text))
|