Spaces:
Runtime error
Runtime error
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"
|