Spaces:

Salmaelbarbori
/

thinkai_um6p

Runtime error

Salmaelbarbori commited on May 7, 2023

Commit

d9fad4e

1 Parent(s): fe2e91a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def translation(text):
     #inference
     inputs = tokenizer("translate to arabic : " + text, return_tensors="pt")
     output = model.generate(**inputs)
-    output = tokenizer.decode(output.cpu().numpy()[0])
     return output

     #inference
     inputs = tokenizer("translate to arabic : " + text, return_tensors="pt")
     output = model.generate(**inputs)
+    output = tokenizer.decode(output.cpu().numpy()[0], skip_special_tokens=True)
     return output