Spaces:
Running
Running
eric sali
commited on
Commit
·
dbd02c3
1
Parent(s):
a0e9950
Add application file
Browse files
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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|