add loader

app.py

@@ -5,11 +5,14 @@ import wave
 import os
 from huggingface_hub import InferenceClient, login
 import logging
+import torchaudio
 
-    # Configurer les logs de debug
+# Configurer les logs de debug
 logging.basicConfig(level=logging.INFO)
 
-
+# Initialisation des variables pour le modèle et le processeur
+model = None
+processor = None
 
 UPLOAD_FOLDER = "audio_files"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
@@ -24,8 +27,23 @@ def serve_react_app():
 def serve_static_files(path):
     return send_from_directory("dist", path)
 
+@app.route("/load_model", methods=["POST"])
+def load_model():
+    global model, processor
+    load = request.json.get("load", False)
+    
+    if load:
+        processor = AutoProcessor.from_pretrained("model_name")  # Remplacez "model_name" par le nom de votre modèle
+        model = SeamlessM4Tv2Model.from_pretrained("model_name")  # Remplacez "model_name" par le nom de votre modèle
+        return jsonify({"message": "Modèle chargé avec succès."}), 200
+    else:
+        return jsonify({"message": "Le modèle n'est pas chargé."}), 200
+
 @app.route("/record", methods=["POST"])
 def record_audio():
+    if model is None or processor is None:
+        return jsonify({"error": "Le modèle n'est pas chargé."}), 400
+    
     file = request.files['audio']
     filename = os.path.join(UPLOAD_FOLDER, file.filename)
     file.save(filename)