Ajouter le script Gradio et les dépendances

app.py

@@ -1,27 +1,24 @@
 import os
+
 import gradio as gr
 from qdrant_client import QdrantClient
 from transformers import ClapModel, ClapProcessor
-
-# Charger les variables d'environnement
 QDRANT_URL = os.getenv("QDRANT_URL")
 QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
-
-# Vérifiez si les variables d'environnement sont définies
-
-# Chargement de la base de données Qdrant en local
-client = QdrantClient(QDRANT_URL, api_key=QDRANT_API_KEY)
+# Loading the Qdrant DB in local ###################################################################
+client = QdrantClient(QDRANT_URL, QDRANT_API_KEY)
 print("[INFO] Client created...")
 
-# Chargement du modèle
+# loading the model
 print("[INFO] Loading the model...")
 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_inputs = processor(text=query, return_tensors="pt")
     text_embed = model.get_text_features(**text_inputs)[0]
@@ -38,12 +35,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 d'avoir des objets différents
+    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()