Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import Tacotron2Processor, Tacotron2ForConditionalGeneration
+from transformers import FastSpeechForConditionalGeneration, Wav2Vec2Processor
 
 # تحميل النموذج والمعالج
-processor = Tacotron2Processor.from_pretrained("facebook/tacotron2")
-model = Tacotron2ForConditionalGeneration.from_pretrained("facebook/tacotron2")
+model = FastSpeechForConditionalGeneration.from_pretrained("facebook/fastspeech2-en-ljspeech")
+processor = Wav2Vec2Processor.from_pretrained("facebook/fastspeech2-en-ljspeech")
 
 def tts(text):
     inputs = processor(text, return_tensors="pt")
@@ -11,5 +11,5 @@ def tts(text):
     return processor.decode(speech[0])
 
 # واجهة Gradio
-iface = gr.Interface(fn=tts, inputs="text", outputs="audio", title="Najdi TTS Model")
+iface = gr.Interface(fn=tts, inputs="text", outputs="audio", title="FastSpeech2 Najdi TTS Model")
 iface.launch()