Spaces:
Build error
Build error
Ajout
Browse files
app.py
CHANGED
|
@@ -12,9 +12,7 @@ load_dotenv()
|
|
| 12 |
QDRANT_URL = os.getenv('QDRANT_URL')
|
| 13 |
QDRANT_KEY = os.getenv('QDRANT_KEY')
|
| 14 |
|
| 15 |
-
|
| 16 |
-
print(f"QDRANT_URL: {QDRANT_URL}")
|
| 17 |
-
print(f"QDRANT_KEY: {QDRANT_KEY}")
|
| 18 |
|
| 19 |
try:
|
| 20 |
# Vérifier que les variables sont correctement récupérées
|
|
@@ -36,22 +34,14 @@ try:
|
|
| 36 |
|
| 37 |
def sound_search(query):
|
| 38 |
try:
|
| 39 |
-
print(f"[DEBUG] Received query: {query}")
|
| 40 |
text_inputs = processor(text=query, return_tensors="pt")
|
| 41 |
-
print(f"[DEBUG] Text inputs: {text_inputs}")
|
| 42 |
text_embed = model.get_text_features(**text_inputs)[0]
|
| 43 |
-
print(f"Text embed: {text_embed}")
|
| 44 |
|
| 45 |
hits = client.search(
|
| 46 |
collection_name="demo_spaces_db",
|
| 47 |
query_vector=text_embed.tolist(),
|
| 48 |
limit=max_results,
|
| 49 |
)
|
| 50 |
-
print(f"Search results: {hits}")
|
| 51 |
-
print([
|
| 52 |
-
hit.payload['audio_s3url']
|
| 53 |
-
|
| 54 |
-
for hit in hits])
|
| 55 |
return [
|
| 56 |
gr.Audio(
|
| 57 |
hit.payload['audio_s3url'],
|
|
|
|
| 12 |
QDRANT_URL = os.getenv('QDRANT_URL')
|
| 13 |
QDRANT_KEY = os.getenv('QDRANT_KEY')
|
| 14 |
|
| 15 |
+
|
|
|
|
|
|
|
| 16 |
|
| 17 |
try:
|
| 18 |
# Vérifier que les variables sont correctement récupérées
|
|
|
|
| 34 |
|
| 35 |
def sound_search(query):
|
| 36 |
try:
|
|
|
|
| 37 |
text_inputs = processor(text=query, return_tensors="pt")
|
|
|
|
| 38 |
text_embed = model.get_text_features(**text_inputs)[0]
|
|
|
|
| 39 |
|
| 40 |
hits = client.search(
|
| 41 |
collection_name="demo_spaces_db",
|
| 42 |
query_vector=text_embed.tolist(),
|
| 43 |
limit=max_results,
|
| 44 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
return [
|
| 46 |
gr.Audio(
|
| 47 |
hit.payload['audio_s3url'],
|