Update app.py

app.py

@@ -1,14 +1,14 @@
 import os
-import time
 import gradio as gr
 import torch
 import numpy as np
 import tempfile
 import librosa
-import whisper
 import re
 from gtts import gTTS
 from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
+import whisper
+import sentencepiece  # Ensure SentencePiece is imported
 
 # Ensure correct device setting
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -16,7 +16,7 @@ print(f"Using device: {device}")
 
 # Load models with error handling
 try:
-    whisper_model = whisper.load_model("small", device=device)
+    whisper_model = whisper.load_model("small")
 except Exception as e:
     print(f"Failed to load Whisper model: {e}")
     whisper_model = None