Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import numpy as np
+from transformers import pipeline, VitsTokenizer, VitsModel, set_seed
 import torch
 import io
 import soundfile as sf
@@ -14,7 +15,7 @@ decode_cfg.beam.beam_size = 1
 canary_model.change_decoding_strategy(decode_cfg)
 
 # Initialize LLM pipeline
-generator = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
+generator = pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)
 
 # Initialize TTS tokenizer and model
 tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-eng")