BeebekBhz's picture
Update app.py
83daf67 verified
raw
history blame contribute delete
759 Bytes
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)