Pavankalyan commited on
Commit
3d99fbb
·
1 Parent(s): 16f7e5e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
+ tokenizer = AutoTokenizer.from_pretrained("salesken/translation-hi-en")
4
+ model = AutoModelForSeq2SeqLM.from_pretrained("salesken/translation-hi-en")
5
+
6
+ def greet(name):
7
+ hin_snippet = name
8
+ inputs = tokenizer.encode(
9
+ hin_snippet, return_tensors="pt",padding=True,max_length=512,truncation=True)
10
+ outputs = model. Generate(inputs, max_length=128, num_beams=None, early_stopping=True)
11
+ translated = tokenizer. Decode(outputs[0]).replace('<pad>',"").strip().lower()
12
+ return translated
13
+
14
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
15
+ iface.launch()