more debug

app.py

@@ -94,7 +94,9 @@ def infer(code):
         early_stopping=False,
         pad_token_id=0,
         eos_token_id=0,
-    )[0]
+    )
+    print(f"Pre Var output: {var_output}")
+    var_output = var_output[0]
     var_output = tokenizer.decode(
         var_output[var_input_ids.size(1) :],
         skip_special_tokens=True,