eric sali commited on
Commit
53edd48
·
1 Parent(s): 7b1e380

Add application file

Browse files
Files changed (1) hide show
  1. app.py +23 -4
app.py CHANGED
@@ -1,10 +1,6 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
-
5
- ##pip install gradio torch transformers
6
- ##pip install huggingface_hub
7
-
8
  tokenizer = AutoTokenizer.from_pretrained("t5-small")
9
  model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
10
 
@@ -31,3 +27,26 @@ translator = gr.Interface(fn=translate_text,
31
  )
32
 
33
  gr.Series(generator, translator).launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
 
 
 
 
4
  tokenizer = AutoTokenizer.from_pretrained("t5-small")
5
  model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
6
 
 
27
  )
28
 
29
  gr.Series(generator, translator).launch()
30
+
31
+
32
+ import gradio as gr
33
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
34
+
35
+ tokenizer = AutoTokenizer.from_pretrained("t5-small")
36
+ model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
37
+
38
+ def translate_text(text):
39
+ inputs = tokenizer.encode("translate English to French: " + text, return_tensors="pt")
40
+ outputs = model.generate(inputs, max_length=128, num_beams=4, early_stopping=True)
41
+ translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
42
+ return translated_text
43
+
44
+ translator = gr.Interface(
45
+ fn=translate_text,
46
+ inputs="text",
47
+ outputs="text",
48
+ title="English to French Translator",
49
+ description="Translate English text to French using the T5-small model.",
50
+ )
51
+
52
+ translator.launch()