fixed code

README.md

@@ -48,8 +48,9 @@ input_text = "FIELD SPIN 0 SU2 2 U1 1 FIELD SPIN 1 / 2 SU3 3 SU2 2 U1 1 / 3 HEL
 4) Perform generation:
 
 ```python
-inputs = tokenizer([input_text], return_tensors="pt")
-outputs = model.generate(**inputs, max_length=2048)
+input_text = "FIELD SPIN 0 SU2 2 U1 1 FIELD SPIN 1 / 2 SU3 3 SU2 2 U1 1 / 3 HEL - 1 / 2"
+inputs = tokenizer(input_text, return_tensors="pt")
+outputs = model.generate(inputs['input_ids'], max_length=2048)
 decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
 print("Generated Lagrangian:")
 print(decoded_outputs[0])