Update README.md

README.md

@@ -38,18 +38,20 @@ query_and_docs = f"Domanda: {query} Contesto: {doc}"
 
 model_input = tokenizer(query_and_docs, truncation=True, padding=True, return_tensors="pt")
 
-output = model.generate(input_ids=model_input["input_ids"],
-                                           attention_mask=model_input["attention_mask"],
-                                           min_length=10,
-                                           max_length=256,
-                                           do_sample=False, 
-                                           early_stopping=True,
-                                           num_beams=8,
-                                           temperature=1.0,
-                                           top_k=None,
-                                           top_p=None,
-                                           no_repeat_ngram_size=3,
-                                           num_return_sequences=1)
+output = model.generate(
+  input_ids=model_input["input_ids"],
+  attention_mask=model_input["attention_mask"],
+  min_length=10,
+  max_length=256,
+  do_sample=False, 
+  early_stopping=True,
+  num_beams=8,
+  temperature=1.0,
+  top_k=None,
+  top_p=None,
+  no_repeat_ngram_size=3,
+  num_return_sequences=1
+)
                                            
 tokenizer.batch_decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True)
 ```