Update app.py

app.py

@@ -640,13 +640,13 @@ def translate(txt,to_lang="en",from_lang=False):
     if(from_lang == to_lang):
         log(f'RET translate with txt as {txt}')
         return txt
-    inputs = tokenizer.encode(f"translate {from_lang} to {to_lang}: " + txt, return_tensors="pt", max_length=float('inf'), truncation=False)
+    inputs = tokenizer.encode(f"translate {from_lang} to {to_lang}: " + txt, return_tensors="pt", max_length=float('inf'), truncation=False).input_ids
     chunks_length = math.ceil(get_tensor_length(inputs) / 512)
     ret = ""
     for index in range(chunks_length):
         ret = ret + ("" if ret == "" else " ") + tokenizer.decode(
             model.generate(
-                torch.tensor([list(inputs[0][ index*512:index*512+512 ])]).input_ids
+                torch.tensor([list(inputs[0][ index*512:index*512+512 ])])
             )[0], skip_special_tokens=True
         )
     log(f'RET translate with ret as {ret}')