shreyasvaidya commited on
Commit
f52e5d3
·
verified ·
1 Parent(s): a1b6a1f

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -11,12 +11,6 @@ from transformers import (
11
  AutoTokenizer,
12
  )
13
  from IndicTransToolkit import IndicProcessor
14
- model_name = "ai4bharat/indictrans2-indic-en-1B"
15
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
16
-
17
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name, trust_remote_code=True)
18
-
19
- ip = IndicProcessor(inference=True)
20
 
21
 
22
 
@@ -27,6 +21,13 @@ DEVICE = "cpu"
27
  # Initialize the OCR object for text detection and recognition
28
  ocr = OCR(device="cpu", verbose=False)
29
  def translate_en_hin(given_str):
 
 
 
 
 
 
 
30
  model = model.to(DEVICE)
31
  model.eval()
32
  src_lang, tgt_lang = "eng_Latn", "hin_Deva"
 
11
  AutoTokenizer,
12
  )
13
  from IndicTransToolkit import IndicProcessor
 
 
 
 
 
 
14
 
15
 
16
 
 
21
  # Initialize the OCR object for text detection and recognition
22
  ocr = OCR(device="cpu", verbose=False)
23
  def translate_en_hin(given_str):
24
+ model_name = "ai4bharat/indictrans2-en-indic-1B"
25
+ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
26
+
27
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name, trust_remote_code=True)
28
+
29
+ ip = IndicProcessor(inference=True)
30
+
31
  model = model.to(DEVICE)
32
  model.eval()
33
  src_lang, tgt_lang = "eng_Latn", "hin_Deva"