leadr64 commited on
Commit
0f1dc48
1 Parent(s): f05ce5a

Ajouter le script Gradio et les dépendances

Browse files
Files changed (1) hide show
  1. test.py +24 -0
test.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ # Charger le modèle de classification audio
5
+
6
+ model_id = "sanchit-gandhi/distilhubert-finetuned-gtzan"
7
+ classifier = pipeline("audio-classification", model=model_id,framework="pt")
8
+
9
+ def classify_audio(audio):
10
+ # Utiliser le modèle de classification audio pour prédire la classe
11
+ result = classifier(audio)
12
+ # Récupérer la classe prédite
13
+ predicted_label = result[0]['label']
14
+ return predicted_label
15
+
16
+ iface = gr.Interface(
17
+ fn=classify_audio,
18
+ inputs=gr.Audio(type="filepath", label="Sélectionnez un fichier audio"),
19
+ outputs="text",
20
+ title="Classification audio",
21
+ description="Ce modèle classe l'audio en différentes catégories."
22
+ )
23
+
24
+ iface.launch()