Berzelius255 commited on
Commit
0605887
Β·
verified Β·
1 Parent(s): beb4559

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  import whisper
3
- from googletrans import Translator
4
  import googletrans
5
  from gtts import gTTS
6
  from io import BytesIO
@@ -156,7 +156,9 @@ def load_ocr_reader(languages):
156
  return easyocr.Reader(languages if languages else ['en'])
157
 
158
  whisper_model = load_whisper_model()
159
- translator = Translator()
 
 
160
 
161
  # Navigation
162
  icons = {"Audio Transcription": "🎀", "Image OCR": "πŸ“„", "Help": "❓"}
@@ -377,9 +379,10 @@ elif selected_tab == "Image OCR":
377
  progress_bar = simulate_progress()
378
 
379
  try:
380
- translation = asyncio.run(translator.translate(
381
- st.session_state.extracted_text, dest=language
382
- ))
 
383
  st.session_state.translated_ocr_text = translation.text
384
  st.success("βœ… Translation complete!")
385
  except Exception as e:
 
1
  import streamlit as st
2
  import whisper
3
+ from deep_translator import GoogleTranslator
4
  import googletrans
5
  from gtts import gTTS
6
  from io import BytesIO
 
156
  return easyocr.Reader(languages if languages else ['en'])
157
 
158
  whisper_model = load_whisper_model()
159
+ def translate_text(text, target_language='en'):
160
+ translator = GoogleTranslator(source='auto', target=target_language)
161
+ return translator.translate(text)
162
 
163
  # Navigation
164
  icons = {"Audio Transcription": "🎀", "Image OCR": "πŸ“„", "Help": "❓"}
 
379
  progress_bar = simulate_progress()
380
 
381
  try:
382
+ #translation = asyncio.run(translator.translate(
383
+ # st.session_state.extracted_text, dest=language
384
+ #))
385
+ translation = translate_text(st.session_state.extracted_text, target_language=language)
386
  st.session_state.translated_ocr_text = translation.text
387
  st.success("βœ… Translation complete!")
388
  except Exception as e: