Update README.md

README.md

@@ -72,14 +72,17 @@ json_arr = [{"name": "order_dinner", "description": "Ordina una cena al ristoran
             ]
 json_string = ' '.join([json.dumps(json_obj) for json_obj in json_arr])
 system_prompt = 'Tu sei un assistenze utile che ha accesso alle seguenti funzioni. Usa le funzioni solo se necessario - \n ' + json_string + ' \n '
-system_prompt
+print(system_prompt)
+
+test_message = [{'role' : 'system' , 'content' : system_prompt2},
+                {'role' : 'user' ,'content' : 'Crea un prodotto di nome AIR size L price 100'}]
 ```
 
 4. Call the model
 
 ```python
 def generate_text():
-    prompt = tokenizer.apply_chat_template(test_message2, tokenize=False)
+    prompt = tokenizer.apply_chat_template(test_message, tokenize=False)
     model_inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
     generated_ids = model.generate(**model_inputs, max_new_tokens=1024)
     return tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]