eric sali commited on
Commit
dbd02c3
·
1 Parent(s): a0e9950

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -20
app.py CHANGED
@@ -26,23 +26,4 @@ translator = gr.Interface(fn=translate_text,
26
  description="Translate English speech to French text using the T5-small model.",
27
  )
28
 
29
- gr.Series(generator, translator).launch()
30
-
31
- tokenizer = AutoTokenizer.from_pretrained("t5-small")
32
- model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
33
-
34
- def translate_text(text):
35
- inputs = tokenizer.encode("translate English to French: " + text, return_tensors="pt")
36
- outputs = model.generate(inputs, max_length=128, num_beams=4, early_stopping=True)
37
- translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
38
- return translated_text
39
-
40
- translator = gr.Interface(
41
- fn=translate_text,
42
- inputs="text",
43
- outputs="text",
44
- title="English to French Translator",
45
- description="Translate English text to French using the T5-small model.",
46
- )
47
-
48
- translator.launch()
 
26
  description="Translate English speech to French text using the T5-small model.",
27
  )
28
 
29
+ gr.Series(generator, translator).launch()