Spaces:
Sleeping
Sleeping
Ajout
Browse files
app.py
CHANGED
@@ -12,6 +12,9 @@ load_dotenv()
|
|
12 |
QDRANT_URL = os.getenv('QDRANT_URL')
|
13 |
QDRANT_KEY = os.getenv('QDRANT_KEY')
|
14 |
|
|
|
|
|
|
|
15 |
|
16 |
try:
|
17 |
# Vérifier que les variables sont correctement récupérées
|
@@ -33,15 +36,22 @@ try:
|
|
33 |
|
34 |
def sound_search(query):
|
35 |
try:
|
|
|
36 |
text_inputs = processor(text=query, return_tensors="pt")
|
|
|
37 |
text_embed = model.get_text_features(**text_inputs)[0]
|
|
|
38 |
|
39 |
hits = client.search(
|
40 |
collection_name="demo_spaces_db",
|
41 |
query_vector=text_embed.tolist(),
|
42 |
limit=max_results,
|
43 |
)
|
|
|
|
|
|
|
44 |
|
|
|
45 |
return [
|
46 |
gr.Audio(
|
47 |
hit.payload['audio_s3url'],
|
@@ -56,6 +66,7 @@ try:
|
|
56 |
gr.Markdown("# Sound search database")
|
57 |
inp = gr.Textbox(placeholder="What sound are you looking for?")
|
58 |
out = [gr.Audio(label=f"{x}") for x in range(max_results)]
|
|
|
59 |
inp.submit(sound_search, inputs=inp, outputs=out)
|
60 |
|
61 |
demo.launch()
|
|
|
12 |
QDRANT_URL = os.getenv('QDRANT_URL')
|
13 |
QDRANT_KEY = os.getenv('QDRANT_KEY')
|
14 |
|
15 |
+
# Vérifier les valeurs récupérées
|
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 |
|
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'],
|
|
|
66 |
gr.Markdown("# Sound search database")
|
67 |
inp = gr.Textbox(placeholder="What sound are you looking for?")
|
68 |
out = [gr.Audio(label=f"{x}") for x in range(max_results)]
|
69 |
+
print(sound_search("fun"))
|
70 |
inp.submit(sound_search, inputs=inp, outputs=out)
|
71 |
|
72 |
demo.launch()
|