Update README.md

README.md

@@ -57,13 +57,16 @@ from google import genai
 # Load model directly from HF
 model = AutoModel.from_pretrained("govtech/lionguard-2.1", trust_remote_code=True)
 
+# Text to classify
+texts = ["hello", "world"]
+
 # Get embeddings (users to input their own Gemini API key)
 client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
-result = client.models.embed_content(
+response = client.models.embed_content(
     model="gemini-embedding-001",
     contents=texts
 )
-embeddings = np.array([emb.values for emb in result.embeddings])
+embeddings = np.array([emb.values for emb in response.embeddings])
 
 # Run inference
 results = model.predict(embeddings)