Update README.md

README.md

@@ -30,17 +30,28 @@ It leverages OpenAI’s `text-embedding-3-large` with a multi-head classifier to
 ---
 
 # Usage
-1. Install packages
-```bash
-pip install -r requirements.txt
-```
-
-2. Set your OpenAI key
-```bash
-export OPENAI_API_KEY=sk-...
-```
-
-3. Run inference on an array of texts
-```bash
-python inference.py "['Text 1', 'Text 2']"
+
+```python
+import os
+import numpy as np
+from transformers import AutoModel
+from openai import OpenAI
+
+# Load model directly from HF
+model = AutoModel.from_pretrained(
+    "govtech/lionguard-2", 
+    trust_remote_code=True
+    )
+
+# Get OpenAI embeddings (users to input their own OpenAI API key)
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+response = client.embeddings.create(
+    input="Hello, world!", # users to input their own text
+    model="text-embedding-3-large",
+    dimensions=1536 # dimensions of the embedding
+    )
+embeddings = np.array([data.embedding for data in response.data])
+
+# Run LionGuard 2
+results = model.predict(embeddings)
 ```