Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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': '
|
| 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 |
-
|
| 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)
|