Update app.py

app.py

@@ -6,11 +6,8 @@ import spaces
 import gc 
 import torch
 
-from gradio_client import Client
-import pyarabic.araby as araby
-
 # Create the necessary directories
-os.makedirs('.gradio/cached_examples/17', exist_ok=True)
+# os.makedirs('.gradio/cached_examples/17', exist_ok=True)
 
 
 def get_model_name(language):
@@ -45,7 +42,7 @@ def load_model(model_name):
     
     return generator
 
-client = Client("MohamedRashad/arabic-auto-tashkeel")
+
 @spaces.GPU
 def generate_kids_story(character, setting, language):
     model_name = get_model_name(language)
@@ -61,9 +58,6 @@ def generate_kids_story(character, setting, language):
 
     messages = [{"role": "user", "content": prompt}]
     output = generator(messages)
-    
-    text = output[0]["generated_text"]
-    
     # Delete model and associated objects 
     del generator
     # Run garbage collection
@@ -71,13 +65,8 @@ def generate_kids_story(character, setting, language):
     # Empty CUDA cache
     torch.cuda.empty_cache()
 
-    # Remove diacritics from Arabic text then add tashkeel
-    if language == "Arabic":
-        text = client.predict(
-        		input_text=araby.strip_diacritics(text),
-        		api_name="/infer_shakkala")
     
-    return text
+    return output[0]["generated_text"]
 
 css_style = """
     body {