amitkot commited on
Commit
25af11a
1 Parent(s): 891562c

feat: change to en->he

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,12 +2,12 @@ import gradio as gr
2
 
3
  from transformers import pipeline
4
 
5
- pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
6
 
7
  def predict(text):
8
  return pipe(text)[0]["translation_text"]
9
 
10
- title = "Hebrew to English Translation"
11
 
12
  iface = gr.Interface(
13
  fn=predict,
 
2
 
3
  from transformers import pipeline
4
 
5
+ pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-he")
6
 
7
  def predict(text):
8
  return pipe(text)[0]["translation_text"]
9
 
10
+ title = "English to Hebrew Translation"
11
 
12
  iface = gr.Interface(
13
  fn=predict,