Update README.md

README.md

@@ -28,14 +28,7 @@ license: other
 
 ## Usage
 
-`StableLM 2 Zephyr 1.6B` uses the following instruction format:
-```
-<|user|>
-Which famous math number begins with 1.6 ...?<|endoftext|>
-<|assistant|>
-The number you are referring to is 1.618033988749895. This is the famous value known as the golden ratio<|endoftext|>
-```
-
+`StableLM 2 Chat` uses the following instruction ChatML format
 This format is also available through the tokenizer's `apply_chat_template` method:
 
 ```python
@@ -65,6 +58,69 @@ tokens = model.generate(
 print(tokenizer.decode(tokens[0], skip_special_tokens=False))
 ```
 
+StableLM 2 Chat also supports function call usage this is an example how you can use it:
+```python
+system_prompt_func = """\
+You are a helpful assistant with access to the following functions. You must use them if required -\n
+[
+  {
+    "type": "function",
+    "function": {
+      "name": "TextToImage",
+      "description": "This function able to creating, drawing, or illustrating an image from a text prompt.",
+      "parameters": {
+        "type": "object",
+        "properties": {
+          "prompt": {
+            "type": "string",
+            "description": "The description of image that user wanto to create."
+          }
+        },
+        "required": [
+          "prompt"
+        ]
+      }
+    }
+    },
+    {
+        "type": "function",
+        "function": {
+        "name": "EditImage",
+        "description": "This is capable of changing, editing, adjusting, or modifying an image by describing changes to the image through a text prompt.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+            "prompt": {
+                "type": "string",
+                "description": "The instruction used to edit the image."
+            }
+            },
+            "required": [
+                "prompt"
+            ]
+        }
+        }
+    }
+]
+"""
+messages = [{'role': 'system', 'content': system_prompt_func}, "user": "Help me to generate a picture of Eiffel Tower in the night!"]
+inputs = tokenizer.apply_chat_template(
+    prompt,
+    add_generation_prompt=True,
+    return_tensors='pt'
+)
+
+tokens = model.generate(
+    inputs.to(model.device),
+    max_new_tokens=1024,
+    temperature=0.5,
+    do_sample=True
+)
+
+print(tokenizer.decode(tokens[0], skip_special_tokens=False))
+```
+
+
 ## Model Details
 
 * **Developed by**: [Stability AI](https://stability.ai/)