Update app.py

app.py

@@ -1,18 +1,34 @@
 import re
-from gender_prediction import get_gender
-import gradio as gr
 import torch
+import gradio as gr
+from gender_prediction import get_gender
 
+# Список аудиофайлов
+audio_files = [
+    "https://huggingface.co/spaces/danielwm994/t1/resolve/main/1.mp3",
+    "https://huggingface.co/spaces/danielwm994/t1/resolve/main/2.mp3",
+    "https://huggingface.co/spaces/danielwm994/t1/resolve/main/3.mp3",
+    "https://huggingface.co/spaces/danielwm994/t1/resolve/main/4.mp3",
+    "https://huggingface.co/spaces/danielwm994/t1/resolve/main/5.mp3"
+]
 
-def app(voice):
-  model_name_or_path = "alefiury/wav2vec2-large-xlsr-53-gender-recognition-librispeech"
-  audio_paths = [voice]
-  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-  predicted_label = get_gender(model_name_or_path, audio_paths, device)
-  gender=re.search("female|male",predicted_label)
-  return gender.string
+def process_audio():
+    model_name_or_path = "alefiury/wav2vec2-large-xlsr-53-gender-recognition-librispeech"
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    results = []
+    
+    for voice in audio_files:
+        predicted_label = get_gender(model_name_or_path, [voice], device)
+        gender = re.search("female|male", predicted_label)
+        results.append({"url": voice, "gender": gender.group(0) if gender else "unknown"})
+    
+    return results
 
+interface = gr.Interface(
+    fn=process_audio,
+    inputs=[],
+    outputs=gr.JSON(label="Results"),
+    live=False
+)
 
-interface=gr.Interface(fn=app,inputs=[gr.components.Audio(type="filepath",sources="upload",label="upload voice")],
-                       outputs=[gr.components.Textbox(label="your result")])
-interface.launch(debug=True)
+interface.launch(debug=True)