jjuarez commited on
Commit
c6f5946
·
1 Parent(s): 657dbb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -1,15 +1,13 @@
1
  import gradio as gr
2
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
 
4
  tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_translation")
5
-
6
  model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_translation")
7
 
8
-
9
  def translate(text):
10
- input_ids = translation_tokenizer.encode(text, return_tensors="pt")
11
- output_ids = translation_model.generate(input_ids)
12
- output_text = translation_tokenizer.decode(output_ids[0], skip_special_tokens=True)
13
  return output_text
14
 
15
  iface_translate = gr.Interface(
@@ -21,4 +19,4 @@ iface_translate = gr.Interface(
21
  theme="compact"
22
  )
23
 
24
- iface_translate.launch()
 
1
  import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_translation")
 
5
  model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_translation")
6
 
 
7
  def translate(text):
8
+ input_ids = tokenizer.encode(text, return_tensors="pt")
9
+ output_ids = model.generate(input_ids)
10
+ output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
11
  return output_text
12
 
13
  iface_translate = gr.Interface(
 
19
  theme="compact"
20
  )
21
 
22
+ iface_translate.launch()