Anupam251272 commited on
Commit
5c73c9b
·
verified ·
1 Parent(s): 7dd66b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  import gradio as gr
4
  import numpy as np
5
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
6
- from googletrans import Translator
7
  from gtts import gTTS
8
  import librosa
9
  import tempfile
@@ -19,9 +19,6 @@ class RealTimeTranslator:
19
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
20
  self.model = self.model.to(self.device)
21
 
22
- # Initialize translator
23
- self.translator = Translator()
24
-
25
  # Supported languages
26
  self.languages = {
27
  'en': 'English',
@@ -52,8 +49,8 @@ class RealTimeTranslator:
52
  def translate_text(self, text, source_lang, target_lang):
53
  """Translate text using Google Translate"""
54
  try:
55
- translation = self.translator.translate(text, src=source_lang, dest=target_lang)
56
- return translation.text
57
  except Exception as e:
58
  return f"Error in translation: {str(e)}"
59
 
 
3
  import gradio as gr
4
  import numpy as np
5
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
6
+ from deep_translator import GoogleTranslator # Replaced googletrans with deep-translator
7
  from gtts import gTTS
8
  import librosa
9
  import tempfile
 
19
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
20
  self.model = self.model.to(self.device)
21
 
 
 
 
22
  # Supported languages
23
  self.languages = {
24
  'en': 'English',
 
49
  def translate_text(self, text, source_lang, target_lang):
50
  """Translate text using Google Translate"""
51
  try:
52
+ translation = GoogleTranslator(source=source_lang, target=target_lang).translate(text)
53
+ return translation
54
  except Exception as e:
55
  return f"Error in translation: {str(e)}"
56