File size: 759 Bytes
ef201a8
 
 
7da6382
ef201a8
7da6382
ef201a8
 
7da6382
 
 
ef201a8
036f632
ef201a8
036f632
83daf67
 
 
 
 
 
036f632
 
 
 
 
83daf67
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import os
import subprocess

# Install PyTorch & Transformers if missing
try:
    import torch
    import transformers
except ImportError:
    subprocess.run(["pip", "install", "torch", "transformers"], check=True)
    import torch  # Re-import after installation
    import transformers

from transformers import pipeline
import gradio as gr

# Define source and target language
src_lang = "eng_Latn"  # English
tgt_lang = "npi_Deva"  # Nepali (Devanagari script)

# Load model with language codes
model = pipeline("translation", model="BeebekBhz/Final-en-ne", src_lang=src_lang, tgt_lang=tgt_lang)

def translate(text):
    return model(text)[0]['translation_text']

iface = gr.Interface(fn=translate, inputs="text", outputs="text")
iface.launch(share=True)