Update README.md

README.md

@@ -124,14 +124,21 @@ The accelerated partition is composed of 1,120 nodes with the following specific
 ---
 
 ## How to use
-This section offers examples of how to perform inference using various methods.
 
-### Inference
+To translate with the salamandraTA-2b model, first you need to create a prompt that specifies the source and target languages in this format:
 
-To run inference using Huggingface's AutoModel class on a single sentence you can use the following code.
+```css
+[source_language] sentence \n[target_language]
+```
+
+You can translate between these languages by using their names directly:
+
+Italian, Portuguese, German, English, Spanish, Euskera, Galician, French, Bulgarian, Czech, Lithuanian, Croatian, Dutch, Romanian, Danish, Greek, Finnish, Hungarian, Slovak, Slovenian, Estonian, Polish, Latvian, Swedish, Maltese, Irish, Aranese, Aragonese, Asturian.
 
-<details>
-<summary>Show code</summary>
+
+### Inference
+
+To translate from Spanish to Catalan using Huggingface's AutoModel class on a single sentence you can use the following code:
 
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -154,7 +161,6 @@ input_length = input_ids.shape[1]
 generated_text = tokenizer.decode(output_ids[0, input_length: ], skip_special_tokens=True).strip()
 # Ahir se'n va anar, va agafar les seves coses i es va posar a navegar.
 ```
-</details>
 
 <br>
 
@@ -204,11 +210,14 @@ for i, output in enumerate(outputs):
 
 print("Generated Translations:", results_detokenized)
 
-#Generated Translations: ["Ahir se'n va anar, va agafar les seves coses i es va posar a navegar.", "Es va acomiadar i va decidir batre's en duel amb el mar, i recórrer el món en el seu veler", 'El seu cor va buscar una forma diferent de viure, però les onades li van cridar: Vés amb els altres', 'I es va adormir i la nit li va cridar: On vas, i en els seus somnis va dibuixar gavines, i va pensar: Avui he de tornar.']
-
+#Generated Translations: ["Ahir se'n va anar, va agafar les seves coses i es va posar a navegar.", 
+#"Es va acomiadar i va decidir batre's en duel amb el mar, i recórrer el món en el seu veler", 
+#"El seu cor va buscar una forma diferent de viure, però les onades li van cridar: Vés amb els altres", 
+#"I es va adormir i la nit li va cridar: On vas, i en els seus somnis va dibuixar gavines, i va pensar: Avui he de tornar."]
 ```
 </details>
 
+
 ## Data