Update app.py

app.py

@@ -2,7 +2,7 @@ import tempfile
 
 import edge_tts
 import gradio as gr
-from transformers import pipeline
+from gradio_client import Client
 import pyarabic.araby as araby
 
 language_dict = {
@@ -442,13 +442,16 @@ language_dict = {
   }
 }
 
-pipe = pipeline("text2text-generation", model="mush42/fine-tashkeel")
+client = Client("MohamedRashad/arabic-auto-tashkeel")
 
 async def text_to_speech_edge(text, language_code, speaker, tashkeel_checkbox=False):
 
   # Remove diacritics from Arabic text then add tashkeel
     if language_code == "Arabic" and tashkeel_checkbox:
-      text = pipe(araby.strip_diacritics(text))[0]["generated_text"].strip()
+        text = client.predict(
+        		input_text=araby.strip_diacritics(text),
+        		api_name="/infer_shakkala"
+        )
     
     # Get the voice for the selected language and speaker
     voice = language_dict[language_code][speaker]