Martin van Dijk commited on
Commit
bbad4ca
·
1 Parent(s): ae0fcb1

Update space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ model = pipeline("translation", model="vonjack/opus-mt-mul-en-big")
4
 
5
 
6
  def predict(prompt):
7
- completion = model(prompt)[0]["generated_text"]
8
  return completion
9
 
10
  import gradio as gr
 
4
 
5
 
6
  def predict(prompt):
7
+ completion = model(prompt)[0]["pt"]
8
  return completion
9
 
10
  import gradio as gr