Update app.py

app.py

@@ -42,9 +42,9 @@ def redistribute_codes(row):
 snac_model = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").to("cuda")
 
 # Load the single-speaker language model
-tokenizer = AutoTokenizer.from_pretrained('prithivMLmods/Llama-3B-Mono-Ceylia')
+tokenizer = AutoTokenizer.from_pretrained('Llama-3B-Mono-Jim')
 model = AutoModelForCausalLM.from_pretrained(
-    'prithivMLmods/Llama-3B-Mono-Ceylia', torch_dtype=torch.bfloat16
+    'Llama-3B-Mono-Jim', torch_dtype=torch.bfloat16
 ).cuda()
 
 @spaces.GPU
@@ -52,7 +52,7 @@ def generate_audio(text, temperature, top_p, max_new_tokens):
     """
     Given input text, generate speech audio.
     """
-    speaker = "Ceylia"
+    speaker = "Jim"
     prompt = f'<custom_token_3><|begin_of_text|>{speaker}: {text}<|eot_id|><custom_token_4><custom_token_5><custom_token_1>'
     input_ids = tokenizer(prompt, add_special_tokens=False, return_tensors='pt').to('cuda')
     
@@ -75,8 +75,8 @@ def generate_audio(text, temperature, top_p, max_new_tokens):
 
 # Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Llama-3B-Mono-Ceylia - Single Speaker Audio Generation")
-    gr.Markdown("Generate speech audio using the `prithivMLmods/Llama-3B-Mono-Ceylia` model.")
+    gr.Markdown("# Llama-3B-Mono-Jim - Single Speaker Audio Generation")
+    gr.Markdown("Generate speech audio using the `Llama-3B-Mono-Jim` model.")
     
     with gr.Row():
         text_input = gr.Textbox(lines=4, label="Input Text")
@@ -96,4 +96,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()