Update app.py

app.py

@@ -3,12 +3,12 @@ import torch
 from transformers import pipeline, AutoTokenizer, AutoModel
 import gradio as gr
 
-# Initialiser le pipeline
-pipe = pipeline("feature-extraction", model="ArthurZ/jukebox-1b-lyrics")
+# Initialiser le pipeline avec "ignore_mismatched_sizes=True"
+pipe = pipeline("feature-extraction", model="ArthurZ/jukebox-1b-lyrics", ignore_mismatched_sizes=True)
 
-# Charger le tokenizer et le modèle
-tokenizer = AutoTokenizer.from_pretrained("ArthurZ/jukebox-1b-lyrics")
-model = AutoModel.from_pretrained("ArthurZ/jukebox-1b-lyrics")
+# Charger le tokenizer et le modèle avec "ignore_mismatched_sizes=True"
+tokenizer = AutoTokenizer.from_pretrained("ArthurZ/jukebox-1b-lyrics", ignore_mismatched_sizes=True)
+model = AutoModel.from_pretrained("ArthurZ/jukebox-1b-lyrics", ignore_mismatched_sizes=True)
 
 # Fonction pour traiter l'audio
 def generate_audio(text):