Update app.py

app.py

@@ -96,8 +96,14 @@ from scipy.io.wavfile import write
 from io import BytesIO
 
 # Load the tokenizer and model
-tokenizer = SeamlessM4TTokenizer.from_pretrained("facebook/seamless-m4t-v2-large")
-model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large")
+# tokenizer = SeamlessM4TTokenizer.from_pretrained("facebook/seamless-m4t-v2-large")
+# model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large")
+
+# Load model directly
+from transformers import AutoProcessor, AutoModelForTextToSpectrogram
+
+processor = AutoProcessor.from_pretrained("Beehzod/speecht5_finetuned_uz_customData")
+model = AutoModelForTextToSpectrogram.from_pretrained("Beehzod/speecht5_finetuned_uz_customData")
 
 # Set the device (CUDA if available, else CPU)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")