emirkaanozdemr commited on
Commit
eeb6c8d
·
verified ·
1 Parent(s): 5cc43db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import streamlit as st
2
  from tensorflow.keras import layers, Model
3
  import tensorflow as tf
 
4
  from tensorflow.keras.models import load_model
5
  from rembg import remove
6
  import numpy as np
@@ -13,7 +14,6 @@ from keras import backend as K
13
  from keras.saving import register_keras_serializable
14
  from deskew import determine_skew
15
  import cv2
16
- from googletrans import Translator
17
  import torch
18
  from diffusers import StableDiffusionImg2ImgPipeline
19
  from transformers import LlamaForCausalLM, PreTrainedTokenizerFast, pipeline
@@ -208,8 +208,8 @@ torch.cuda.empty_cache()
208
  language = st.selectbox("Select Language", list(languages.keys()))
209
 
210
  if language:
211
- translator = Translator()
212
- tr_list_tr = [translator.translate(text, dest=languages[language]).text for text in tr_list]
213
 
214
  st.title(tr_list_tr[0])
215
 
 
1
  import streamlit as st
2
  from tensorflow.keras import layers, Model
3
  import tensorflow as tf
4
+ from deep_translator import GoogleTranslator
5
  from tensorflow.keras.models import load_model
6
  from rembg import remove
7
  import numpy as np
 
14
  from keras.saving import register_keras_serializable
15
  from deskew import determine_skew
16
  import cv2
 
17
  import torch
18
  from diffusers import StableDiffusionImg2ImgPipeline
19
  from transformers import LlamaForCausalLM, PreTrainedTokenizerFast, pipeline
 
208
  language = st.selectbox("Select Language", list(languages.keys()))
209
 
210
  if language:
211
+ translator = GoogleTranslator(source='auto', target=languages[language])
212
+ tr_list_tr = [translator.translate(text) for text in tr_list]
213
 
214
  st.title(tr_list_tr[0])
215