leadr64 commited on
Commit
6711b9e
1 Parent(s): ecf02ba
Files changed (1) hide show
  1. app.py +11 -0
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()