Update app.py

app.py

@@ -6,15 +6,17 @@ from xcodec2.modeling_xcodec2 import XCodec2Model
 import torchaudio
 import gradio as gr
 import tempfile
+import os 
 
-llasa_3b ='srinivasbilla/llasa-3b'
+llasa_1b ='SebastianBodza/Kartoffel-1B-v0.1'
 
-tokenizer = AutoTokenizer.from_pretrained(llasa_3b)
+tokenizer = AutoTokenizer.from_pretrained(llasa_1b, token=os.getenv("HF_TOKEN"))
 
 model = AutoModelForCausalLM.from_pretrained(
-    llasa_3b,
+    llasa_1b,
     trust_remote_code=True,
     device_map='cuda',
+    token=os.getenv("HF_TOKEN")
 )
 
 model_path = "srinivasbilla/xcodec2"
@@ -162,9 +164,9 @@ with gr.Blocks() as app_credits:
 with gr.Blocks() as app:
     gr.Markdown(
         """
-# llasa 3b TTS
+# Kartoffel-1B-v0.1 - llasa 1b TTS
 
-This is a local web UI for llasa 3b SOTA(imo) Zero Shot Voice Cloning and TTS model.
+This is a local web UI for my finetune of the llasa 1b SOTA(imo) Zero Shot Voice Cloning and TTS model.
 
 The checkpoints support English and Chinese.