Ajouter le script Gradio et les dépendances

app.py

@@ -1,29 +1,29 @@
-import gradio as gr
-import laion_clap
-from qdrant_client import QdrantClient
 import os
 
-# Utilisez les variables d'environnement pour la configuration
-QDRANT_HOST = os.getenv('QDRANT_HOST', 'localhost')
-QDRANT_PORT = int(os.getenv('QDRANT_PORT', 6333))
+import gradio as gr
+from qdrant_client import QdrantClient
+from transformers import ClapModel, ClapProcessor
 
-# Connexion à Qdrant
-client = QdrantClient(QDRANT_HOST, port=QDRANT_PORT)
+# Loading the Qdrant DB in local ###################################################################
+client = QdrantClient("https://ebe79742-e3ac-4d09-a2c6-63946024cc7a.us-east4-0.gcp.cloud.qdrant.io", api_key="_NnGLuSMH4Qwv-ancoFh88YvzuR7WbyidAorVOVQ_eMCbPhxTb2TSw")
 print("[INFO] Client created...")
 
-# Charger le modèle
+# loading the model
 print("[INFO] Loading the model...")
-model_name = "laion/larger_clap_music"
-model = laion_clap.CLAP_Module(enable_fusion=False)
-model.load_ckpt()  # télécharger le checkpoint préentraîné par défaut
+model_name = "laion/larger_clap_general"
+model = ClapModel.from_pretrained(model_name)
+processor = ClapProcessor.from_pretrained(model_name)
 
-# Interface Gradio
+# Gradio Interface #################################################################################
 max_results = 10
 
+
 def sound_search(query):
-    text_embed = model.get_text_embedding([query, ''])[0]  # trick because can't accept singleton
+    text_inputs = processor(text=query, return_tensors="pt")
+    text_embed = model.get_text_features(**text_inputs)[0]
+
     hits = client.search(
-        collection_name="demo_db7",
+        collection_name="demo_spaces_db",
         query_vector=text_embed,
         limit=max_results,
     )
@@ -34,12 +34,13 @@ def sound_search(query):
         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)]  # Nécessaire pour avoir différents objets
+    out = [gr.Audio(label=f"{x}") for x in range(max_results)]  # Necessary to have different objs
     inp.change(sound_search, inp, out)
 
 demo.launch()