Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import torch
-from transformers import AutoProcessor, AutoModel
+from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor
 
 # Load Kokoro TTS Model
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_name = "hexgrad/Kokoro-82M"
-model = AutoModel.from_pretrained(model_name).to(device)
+model = AutoModelForSpeechSeq2Seq.from_pretrained(model_name).to(device)
 processor = AutoProcessor.from_pretrained(model_name)
 
 def text_to_speech(text):