Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -29,7 +29,6 @@ app = FastAPI(title="E-Bikes Semantic Search API",
|
|
29 |
# Request and response models
|
30 |
class SearchRequest(BaseModel):
|
31 |
description: str
|
32 |
-
top_k: int = 3
|
33 |
|
34 |
class BikeMatch(BaseModel):
|
35 |
id: str
|
@@ -182,7 +181,7 @@ async def search_ebikes(request: SearchRequest):
|
|
182 |
# Query Pinecone
|
183 |
results = pinecone_index.query(
|
184 |
vector=query_embedding.tolist(),
|
185 |
-
top_k=
|
186 |
include_metadata=True
|
187 |
)
|
188 |
|
|
|
29 |
# Request and response models
|
30 |
class SearchRequest(BaseModel):
|
31 |
description: str
|
|
|
32 |
|
33 |
class BikeMatch(BaseModel):
|
34 |
id: str
|
|
|
181 |
# Query Pinecone
|
182 |
results = pinecone_index.query(
|
183 |
vector=query_embedding.tolist(),
|
184 |
+
top_k=3,
|
185 |
include_metadata=True
|
186 |
)
|
187 |
|