Update README.md

README.md

@@ -26,93 +26,6 @@ This model excels at Multi-turn Function Calling with tools from `gmail`, `jira`
 
 The code used to generate the dataset can be found [here](https://github.com/prem-research/Funcdex-Synthesizer).
 
-
-# Quickstart
-
-```python
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from peft import PeftModel
-import torch
-import json
-
-# Load model and tokenizer
-base_model_name = "ojus1/Qwen3-1.7B-Instruct"
-model_name = "prem-research/Funcdex-1.7B"
-
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-
-base_model = AutoModelForCausalLM.from_pretrained(
-    base_model_name,
-    torch_dtype="auto",
-    device_map="auto"
-)
-
-model = PeftModel.from_pretrained(
-    base_model,
-    model_name,
-    torch_dtype="auto",
-    device_map="auto"
-)
-
-# Define tools (supports all toolkits)
-tools = [
-    {
-        "type": "function",
-        "function": {
-            "name": "CREATE_SHARED_DRIVE",
-            "description": "Create a new shared drive in Google Drive",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "name": {"type": "string", "description": "Name of the shared drive"},
-                    "requestId": {"type": "string", "description": "Unique request ID"}
-                },
-                "required": ["name", "requestId"]
-            }
-        }
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "CREATE_A_FOLDER",
-            "description": "Create a folder in Google Drive",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "folder_name": {"type": "string", "description": "Name of the folder"},
-                    "parent_id": {"type": "string", "description": "Parent drive or folder ID"}
-                },
-                "required": ["folder_name", "parent_id"]
-            }
-        }
-    }
-]
-
-# Define conversation
-messages = [
-    {"role": "system", "content": "You are a helpful assistant that can help with tasks by using tools."},
-    {"role": "user", "content": "Create a shared drive named 'Partner-Alpha-Integration' with request ID 'req-12345'."}
-]
-
-# Apply chat template with tools
-formatted_input = tokenizer.apply_chat_template(
-    messages,
-    tools=tools,
-    tokenize=False,
-    add_generation_prompt=True
-)
-
-# Tokenize and generate
-input_tokens = tokenizer(formatted_input, return_tensors="pt").to(model.device)
-output = model.generate(**input_tokens, max_new_tokens=256, do_sample=False)
-response = tokenizer.decode(output[0][input_tokens['input_ids'].shape[1]:], skip_special_tokens=True)
-
-print("Response:", response)
-# Expected output includes: <tool_call>{"name": "CREATE_SHARED_DRIVE", "arguments": {"name": "Partner-Alpha-Integration", "requestId": "req-12345"}}</tool_call>
-```
-
-For best results, provide detailed system-prompt to steer the tool-use behaviour.
-
 # Evaluation
 
 
@@ -570,6 +483,92 @@ EM is a strict metric, and penalizes string arguments in function calls that may
 </table>
 
 
+# Quickstart
+
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from peft import PeftModel
+import torch
+import json
+
+# Load model and tokenizer
+base_model_name = "ojus1/Qwen3-1.7B-Instruct"
+model_name = "prem-research/Funcdex-1.7B"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+base_model = AutoModelForCausalLM.from_pretrained(
+    base_model_name,
+    torch_dtype="auto",
+    device_map="auto"
+)
+
+model = PeftModel.from_pretrained(
+    base_model,
+    model_name,
+    torch_dtype="auto",
+    device_map="auto"
+)
+
+# Define tools (supports all toolkits)
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "CREATE_SHARED_DRIVE",
+            "description": "Create a new shared drive in Google Drive",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "name": {"type": "string", "description": "Name of the shared drive"},
+                    "requestId": {"type": "string", "description": "Unique request ID"}
+                },
+                "required": ["name", "requestId"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "CREATE_A_FOLDER",
+            "description": "Create a folder in Google Drive",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "folder_name": {"type": "string", "description": "Name of the folder"},
+                    "parent_id": {"type": "string", "description": "Parent drive or folder ID"}
+                },
+                "required": ["folder_name", "parent_id"]
+            }
+        }
+    }
+]
+
+# Define conversation
+messages = [
+    {"role": "system", "content": "You are a helpful assistant that can help with tasks by using tools."},
+    {"role": "user", "content": "Create a shared drive named 'Partner-Alpha-Integration' with request ID 'req-12345'."}
+]
+
+# Apply chat template with tools
+formatted_input = tokenizer.apply_chat_template(
+    messages,
+    tools=tools,
+    tokenize=False,
+    add_generation_prompt=True
+)
+
+# Tokenize and generate
+input_tokens = tokenizer(formatted_input, return_tensors="pt").to(model.device)
+output = model.generate(**input_tokens, max_new_tokens=256, do_sample=False)
+response = tokenizer.decode(output[0][input_tokens['input_ids'].shape[1]:], skip_special_tokens=True)
+
+print("Response:", response)
+# Expected output includes: <tool_call>{"name": "CREATE_SHARED_DRIVE", "arguments": {"name": "Partner-Alpha-Integration", "requestId": "req-12345"}}</tool_call>
+```
+
+For best results, provide detailed system-prompt to steer the tool-use behaviour.
+
 # License
 
 The models, code and the dataset are licensed under MIT License.