att

Dockerfile

@@ -12,5 +12,5 @@ WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-COPY --chown=user . /app
+COPY --chown=user app.py /app
 CMD ["python", "app.py"]

app.py

@@ -33,8 +33,8 @@ def load_model():
     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
+        processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large")  # Remplacez "model_name" par le nom de votre modèle
+        model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large")  # 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