File size: 719 Bytes
337fc64
3249698
337fc64
 
 
 
 
 
5ca8da0
d0b7d87
337fc64
 
 
 
 
 
 
 
 
 
d0b7d87
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
import gradio as gr
model_ckpt = "booba-uz/english-uzbek-translation_v1"
tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
model = AutoModelForSeq2SeqLM.from_pretrained(model_ckpt)
tokenizer.src_lang = "en"
tokenizer.tgt_lang = "uz"
def translator(input_text):  
    if not input_text.strip():
      return "Iltimos Gap kiriting"
    inputs = tokenizer(input_text, return_tensors="pt")
    outputs = model.generate(**inputs, max_length=256)
    translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
    return translated_text
demo = gr.Interface(
    fn=translator, 
    inputs="text", 
    outputs="text",
)

demo.launch(debug=True)