Update app.py

app.py

@@ -1,5 +1,8 @@
 from __future__ import annotations
 import os
+#download for mecab
+os.system('python -m unidic download')
+
 # we need to compile a CUBLAS version 
 # Or get it from  https://jllllll.github.io/llama-cpp-python-cuBLAS-wheels/
 os.system('CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python==0.2.11')
@@ -25,7 +28,7 @@ import langid
 import uuid
 import emoji
 import pathlib
-import base64
+
 import datetime
 
 from scipy.io.wavfile import write
@@ -52,8 +55,6 @@ import numpy as np
 from gradio_client import Client
 from huggingface_hub import InferenceClient
 
-SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
-
 # This will trigger downloading model
 print("Downloading if not downloaded Coqui XTTS V2")
 
@@ -63,6 +64,7 @@ ModelManager().download_model(model_name)
 model_path = os.path.join(get_user_data_dir("tts"), model_name.replace("/", "--"))
 print("XTTS downloaded")
 
+
 print("Loading XTTS")
 config = XttsConfig()
 config.load_json(os.path.join(model_path, "config.json"))
@@ -78,9 +80,12 @@ model.load_checkpoint(
 model.cuda()
 print("Done loading TTS")
 
-title = "Voice chat with Zephyr and Coqui XTTS"
+#####llm_model = os.environ.get("LLM_MODEL", "mistral") # or "zephyr"
+
+title = "Voice chat with Zephyr/Mistral and Coqui XTTS"
 
-DESCRIPTION = """# Voice chat with Zephyr and Coqui XTTS"""
+DESCRIPTION = """# Voice chat with Zephyr/Mistral and Coqui XTTS"""
+css = """.toast-wrap { display: none !important } """
 
 from huggingface_hub import HfApi
 
@@ -88,9 +93,11 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 # will use api to restart space on a unrecoverable error
 api = HfApi(token=HF_TOKEN)
 
-repo_id = "jbilcke-hf/zephyr-xtts"
-
+# config changes by Julian ---------------
+repo_id = "jbilcke-hf/ai-bedtime-story-server"
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 SENTENCE_SPLIT_LENGTH=250
+# ----------------------------------------
 
 default_system_message = f"""
 You're the storyteller, crafting a short tale for young listeners. Please abide by these guidelines: