Update cloning/clonevoice.py

cloning/clonevoice.py

@@ -6,10 +6,17 @@ from bark.hubert.customtokenizer import CustomTokenizer
 
 import torchaudio
 import torch
+# Charge le modèle sur le CPU
+model = torch.load('chemin_vers_ton_modele.pth', map_location=torch.device('cpu'))
+model = model.eval()
+
+# Si vous avez besoin de déplacer le modèle vers un dispositif spécifique après le chargement
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
+
 import os
 import gradio
 
-
 def clone_voice(audio_filepath, dest_filename, progress=gradio.Progress(track_tqdm=True)):
     # if len(text) < 1:
     #    raise gradio.Error('No transcription text entered!')