Update app.py

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
 from transformers import pipeline
-import torch
 
 # Initialize Hugging Face pipelines
 speech_to_text = pipeline("automatic-speech-recognition", model="openai/whisper-large")
 
-# Use a valid TTS model for text-to-speech (VITS model from Hugging Face)
-text_to_speech = pipeline("text-to-speech", model="espnet/kan-bayashi_ljspeech_vits", device=0)  # Use CPU or GPU (cuda)
+# Use a compatible TTS model for text-to-speech (FastSpeech2)
+text_to_speech = pipeline("text-to-speech", model="facebook/fastspeech2-en-ljspeech", device=0)  # Use CPU or GPU (cuda)
 
 # Function to process speech to text and text to speech
 def process_audio(input_audio):