Update README.md

README.md

@@ -25,7 +25,7 @@ from ispin.ISPIn import ISPIn
 model = ISPIn.from_pretrained('Lutech-AI/I-SPIn')
 sentences = ['Questa è una frase di prova', 'Testando il funzionamento del modello']
 sentence_embeddings = model(sentences)
-print(sentence_embeddings)
+print(sentence_embeddings) # -> torch.Size(2, 768)
 ```
 
 ## Retrieve labels
@@ -42,7 +42,7 @@ predictions = model.predict(
     hypothesis_embeddings,
     one_to_many = True
 )
-print(predictions)
+print(predictions) # -> [[0][0]]
 ```
 
 ## Usage #2 (cloning repo) (deprecated, will be deleted)