ysharma HF staff commited on
Commit
da7d7ed
1 Parent(s): 5cf02ff
Files changed (1) hide show
  1. app.py +2 -11
app.py CHANGED
@@ -5,7 +5,6 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
  import matplotlib.pyplot as plt
6
  import keras_ocr
7
  import cv2
8
- #from cv2 import INPAINT_NS, inpaint, line, cvtColor, COLOR_BGR2RGB, imwrite
9
  import math
10
  import numpy as np
11
  import gradio as gr
@@ -13,7 +12,6 @@ import numpy as np
13
 
14
 
15
  #Support for Hindi, Spanish, French, Arabic, Turish, Gailec/Irish, and German
16
-
17
  #'hindi':
18
  tokenizerhi = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
19
  modelhi = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
@@ -48,25 +46,18 @@ def choose_language(language):
48
  #Loading the tokenizers and trained models
49
  if language == 'hindi':
50
  tokenizer, model = tokenizerhi, modelhi
51
- #modelnm = "Helsinki-NLP/opus-mt-en-hi"
52
  elif language == 'spanish':
53
  tokenizer, model = tokenizeres, modeles
54
- #modelnm = "Helsinki-NLP/opus-mt-en-es"
55
  elif language == 'german':
56
  tokenizer, model = tokenizerde, modelde
57
- #modelnm = "Helsinki-NLP/opus-mt-en-de"
58
  elif language == 'french':
59
  tokenizer, model = tokenizerfr, modelfr
60
- #modelnm = "Helsinki-NLP/opus-mt-en-fr"
61
  elif language == 'turkish':
62
  tokenizer, model = tokenizertrk, modeltrk
63
- #modelnm = "Helsinki-NLP/opus-mt-en-trk"
64
  elif language == 'arabic':
65
  tokenizer, model = tokenizerar, modelar
66
- #modelnm = "Helsinki-NLP/opus-mt-en-ar"
67
  else:
68
  tokenizer, model = tokenizerga, modelga
69
- #modelnm = "Helsinki-NLP/opus-mt-en-ga"
70
  return tokenizer, model
71
 
72
  #Function to translate english text to desired language
@@ -146,7 +137,7 @@ def translate_image(im, language):
146
  #Extract text, translate in your language and format it
147
  extracted_text = text_extract(im)
148
  #font select -- Getting Unicode Text
149
- title_font = ImageFont.truetype('./arial-unicode-ms.ttf',30) #('/content/gdrive/My Drive/sample_images/arial-unicode-ms.ttf',30)
150
  #text to write on image #Example in hindi - Unicode text u"आप जीवन में मिलता हर मौका ले लो, क्योंकि कुछ चीजें केवल एक बार होती हैं. शुभ सुबह"
151
  txt = format_text(language,extracted_text)
152
 
@@ -158,7 +149,7 @@ def translate_image(im, language):
158
 
159
  #Enable writing on image
160
  image_editable = ImageDraw.Draw(new_image)
161
- image_editable.multiline_text((40,40), txt,spacing=2, font=title_font, fill= (0, 0, 0)) #(237, 230, 211)) Whites
162
  return new_image
163
 
164
 
5
  import matplotlib.pyplot as plt
6
  import keras_ocr
7
  import cv2
 
8
  import math
9
  import numpy as np
10
  import gradio as gr
12
 
13
 
14
  #Support for Hindi, Spanish, French, Arabic, Turish, Gailec/Irish, and German
 
15
  #'hindi':
16
  tokenizerhi = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
17
  modelhi = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-hi")
46
  #Loading the tokenizers and trained models
47
  if language == 'hindi':
48
  tokenizer, model = tokenizerhi, modelhi
 
49
  elif language == 'spanish':
50
  tokenizer, model = tokenizeres, modeles
 
51
  elif language == 'german':
52
  tokenizer, model = tokenizerde, modelde
 
53
  elif language == 'french':
54
  tokenizer, model = tokenizerfr, modelfr
 
55
  elif language == 'turkish':
56
  tokenizer, model = tokenizertrk, modeltrk
 
57
  elif language == 'arabic':
58
  tokenizer, model = tokenizerar, modelar
 
59
  else:
60
  tokenizer, model = tokenizerga, modelga
 
61
  return tokenizer, model
62
 
63
  #Function to translate english text to desired language
137
  #Extract text, translate in your language and format it
138
  extracted_text = text_extract(im)
139
  #font select -- Getting Unicode Text
140
+ title_font = ImageFont.truetype('./arial-unicode-ms.ttf',30)
141
  #text to write on image #Example in hindi - Unicode text u"आप जीवन में मिलता हर मौका ले लो, क्योंकि कुछ चीजें केवल एक बार होती हैं. शुभ सुबह"
142
  txt = format_text(language,extracted_text)
143
 
149
 
150
  #Enable writing on image
151
  image_editable = ImageDraw.Draw(new_image)
152
+ image_editable.multiline_text((10,10), txt,spacing=2, font=title_font, fill= (237, 230, 211)) # Text color e.g. (0, 0, 0)) blacks
153
  return new_image
154
 
155