BeastGokul commited on
Commit
c87d667
·
verified ·
1 Parent(s): e5b579c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from PIL import Image
3
  import os
4
  from ocr_tamil.ocr import OCR
5
- from googletrans import Translator
6
 
7
  # Load OCR models
8
  ocr_detect = OCR(detect=True, enable_cuda=False)
@@ -11,7 +11,7 @@ ocr_recognize = OCR(detect=False, enable_cuda=False)
11
  # Language code mapping
12
  lang_codes = {
13
  'English': 'en',
14
- 'Kannada': 'ka',
15
  'Tamil': 'ta',
16
  'Telugu': 'te',
17
  'Hindi': 'hi',
@@ -29,9 +29,7 @@ def predict(image_path, target_language):
29
  texts = [" ".join(texts[0])]
30
  detected_text = texts[0]
31
 
32
- translator = Translator()
33
- translated_text = translator.translate(detected_text, dest=target_language).text
34
-
35
  return detected_text, translated_text
36
  except Exception as e:
37
  return f"Error: {str(e)}", ""
@@ -54,4 +52,4 @@ if uploaded_file:
54
  st.success("Detected Text:")
55
  st.write(detected)
56
  st.success("Translated Text:")
57
- st.write(translated)
 
2
  from PIL import Image
3
  import os
4
  from ocr_tamil.ocr import OCR
5
+ from deep_translator import GoogleTranslator
6
 
7
  # Load OCR models
8
  ocr_detect = OCR(detect=True, enable_cuda=False)
 
11
  # Language code mapping
12
  lang_codes = {
13
  'English': 'en',
14
+ 'Kannada': 'kn',
15
  'Tamil': 'ta',
16
  'Telugu': 'te',
17
  'Hindi': 'hi',
 
29
  texts = [" ".join(texts[0])]
30
  detected_text = texts[0]
31
 
32
+ translated_text = GoogleTranslator(source='auto', target=target_language).translate(detected_text)
 
 
33
  return detected_text, translated_text
34
  except Exception as e:
35
  return f"Error: {str(e)}", ""
 
52
  st.success("Detected Text:")
53
  st.write(detected)
54
  st.success("Translated Text:")
55
+ st.write(translated)