Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import AutoTokenizer,VitsModel
 
 import google.generativeai as genai
 import torch
+import torchaudio
 
 api_key =os.environ.get("id_gmkey")
 token=os.environ.get("key_")
@@ -195,11 +196,13 @@ def   get_answer_ai(text,session_ai):
               session_ai=create_chat_session()
               response = session_ai.send_message(text,stream=True)
               return response,session_ai
+import torchaudio
 
 def   modelspeech(text):
-     audio_bytes = query({
-	"inputs":text   })
-     return audio_bytes
+     audio_bytes = query({"inputs":text   })
+     wav, sr = torchaudio.load(audio_bytes)
+       
+     return sr,wav.squeeze().cpu().numpy()
      with torch.no_grad():
           inputs = tokenizer(text, return_tensors="pt")#.cuda()