Spaces:

ejschwartz
/

resym

Running on Zero

ejschwartz commited on Mar 24

Commit

9a4b368

1 Parent(s): 5708410

oops!

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,7 +119,7 @@ def infer(code):
             eos_token_id=0,
         )[0]
         field_output = tokenizer.decode(
-            field_output[var_input_ids.size(1) :],
             skip_special_tokens=True,
             clean_up_tokenization_spaces=True,
         )

             eos_token_id=0,
         )[0]
         field_output = tokenizer.decode(
+            field_output[field_input_ids.size(1) :],
             skip_special_tokens=True,
             clean_up_tokenization_spaces=True,
         )