Kari commited on
Commit
c818cd3
·
1 Parent(s): 405ef51

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
1
  # Helsinki-NLP/opus-mt-zh-en
2
  import gradio as gr
3
  from transformers import pipeline
 
 
4
 
5
  def en2zh(EnName):
6
- translator = pipeline("translation", model=".\\test-ml-trained", max_time=7)
7
  pre = translator(EnName, )[0]["translation_text"]
8
  return pre
9
 
 
1
  # Helsinki-NLP/opus-mt-zh-en
2
  import gradio as gr
3
  from transformers import pipeline
4
+ import os
5
+ translator = pipeline("translation", model=os.path.join(os.getcwd(),"test-ml-trained"), max_time=5)
6
 
7
  def en2zh(EnName):
 
8
  pre = translator(EnName, )[0]["translation_text"]
9
  return pre
10