leannetanyt commited on
Commit
efdd6d9
·
verified ·
1 Parent(s): cc0fd7b

feat: upload inference script

Browse files
Files changed (1) hide show
  1. inference.py +18 -0
inference.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import numpy as np
3
+ from openai import OpenAI
4
+ from transformers import AutoModel
5
+
6
+ texts = ["Eh you damn stupid lah!", "Have a nice day :)", "This is cool~"]
7
+
8
+ # Load model directly from Hub
9
+ model = AutoModel.from_pretrained("govtech/lionguard-2", trust_remote_code=True)
10
+
11
+ # Get embeddings (users to input their own OpenAI API key)
12
+ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
13
+ response = client.embeddings.create(input=texts, model="text-embedding-3-large")
14
+ embeddings = np.array([data.embedding for data in response.data])
15
+
16
+ # Run inference
17
+ predictions = model.predict(embeddings)
18
+ print(predictions)