File size: 794 Bytes
11c0516 ca47807 11c0516 ca47807 11c0516 ca47807 11c0516 ca47807 11c0516 30eef39 11c0516 30eef39 |
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 28 29 30 |
import os
import subprocess
# Install required dependencies if missing
try:
import torch
import transformers
import sentencepiece
except ImportError:
subprocess.run(["pip", "install", "torch", "transformers", "sentencepiece"], check=True)
import torch # Re-import after installation
import transformers
import sentencepiece
from transformers import pipeline
import gradio as gr
# Define source and target language
src_lang = "ne" # Nepali
tgt_lang = "en" # English
# Load model with language codes
model = pipeline("translation", model="BeebekBhz/Final-ne-en", 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)
|