Spaces:

Salmaelbarbori
/

thinkai_um6p

Runtime error

Salmaelbarbori commited on May 7, 2023

Commit

397fde2

1 Parent(s): d9fad4e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def translation(text):
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
     #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)

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