Spaces:

vilarin
/

LLaMAX3-Translator

Running on Zero

vilarin commited on Jul 25, 2024

Commit

112e28c

verified ·

1 Parent(s): fca0518

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ def translate(
     streamer = TextIteratorStreamer(tokenizer, **{"skip_special_tokens": True, "skip_prompt": True, 'clean_up_tokenization_spaces':False,})
     generate_kwargs = dict(
-        input_ids=inputs,
         streamer=streamer,
         max_length=max_length,
         do_sample=True,

     streamer = TextIteratorStreamer(tokenizer, **{"skip_special_tokens": True, "skip_prompt": True, 'clean_up_tokenization_spaces':False,})
     generate_kwargs = dict(
+        inputs.input_ids,
         streamer=streamer,
         max_length=max_length,
         do_sample=True,