Ajouter le script Gradio et les dépendances

app.py

@@ -1,47 +1,50 @@
 import os
-
 import gradio as gr
 from qdrant_client import QdrantClient
 from transformers import ClapModel, ClapProcessor
+
+# Retrieve Qdrant URL and API key from environment variables
 QDRANT_URL = os.getenv("QDRANT_URL")
 QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
+
+if not QDRANT_URL or not QDRANT_API_KEY:
+    raise ValueError("Please set the QDRANT_URL and QDRANT_API_KEY environment variables")
+
 # Loading the Qdrant DB in local ###################################################################
-client = QdrantClient(QDRANT_URL, QDRANT_API_KEY)
+client = QdrantClient(QDRANT_URL, api_key=QDRANT_API_KEY)
 print("[INFO] Client created...")
 
 # loading the model
 print("[INFO] Loading the model...")
-model_name = "laion/larger_clap_general"
+model_name = "laion/clap-large-v2"
 model = ClapModel.from_pretrained(model_name)
 processor = ClapProcessor.from_pretrained(model_name)
 
 # Gradio Interface #################################################################################
 max_results = 10
 
-
 def sound_search(query):
     text_inputs = processor(text=query, return_tensors="pt")
     text_embed = model.get_text_features(**text_inputs)[0]
 
     hits = client.search(
         collection_name="demo_spaces_db",
-        query_vector=text_embed,
+        query_vector=text_embed.tolist(),  # Convert tensor to list
         limit=max_results,
     )
     return [
         gr.Audio(
-            hit.payload['audio_path'],
+            value=hit.payload['audio_path'],
             label=f"style: {hit.payload['style']} -- score: {hit.score}")
         for hit in hits
     ]
 
-
 with gr.Blocks() as demo:
     gr.Markdown(
         """# Sound search database """
     )
     inp = gr.Textbox(placeholder="What sound are you looking for ?")
     out = [gr.Audio(label=f"{x}") for x in range(max_results)]  # Necessary to have different objs
-    inp.change(sound_search, inp, out)
+    inp.change(fn=sound_search, inputs=inp, outputs=out)
 
 demo.launch()