Update README.md

README.md

@@ -1,6 +1,11 @@
 ---
 library_name: transformers
-tags: []
+license: mit
+datasets:
+- MattCoddity/dockerNLcommands
+language:
+- en
+new_version: Bruce1489/Llama3.2-docker-command-v2
 ---
 
 # Model Card for Model ID
@@ -38,8 +43,35 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
 ### Direct Use
+```{python}
+import torch
+from transformers import pipeline
+
+pipe = pipeline(
+    "text-generation",
+    model="Bruce1489/Llama3.2-docker-command-v2",
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
+    # tokenizer = tokenizer
+)
+messages = [
+    {"role": "system", "content": "translate this sentence in docker command"},
+    {"role": "user", "content": "Please show me the Docker containers that have exited and are related to the mongo image."},
+]
+
+outputs = pipe(
+    messages,
+    max_new_tokens=256,
+    temperature = 0.5,
+    top_p  = 0.9,
+    top_k = 10,
+    do_sample = True,
+    repetition_penalty = 1.2
+)
+print(outputs[0]["generated_text"][-1]['content'])
+
+```
 
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
 
 [More Information Needed]
 
@@ -74,6 +106,11 @@ Use the code below to get started with the model.
 [More Information Needed]
 
 ## Training Details
+```{python}
+
+```
+
+
 
 ### Training Data