Upload folder using huggingface_hub
Browse files
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"
|