Update inference.py

inference.py

@@ -1,4 +1,5 @@
 import openai
+from openai import OpenAI
 from transformers import BertTokenizer
 from evo_model import EvoTransformerForClassification
 import torch
@@ -11,18 +12,17 @@ model.eval()
 # Tokenizer (BERT-compatible)
 tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
 
-# Set OpenAI key (assumes you have it set as ENV VAR or replace directly)
-openai.api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 def query_gpt35(prompt):
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[{"role": "user", "content": prompt}],
             max_tokens=50,
             temperature=0.3,
         )
-        return response['choices'][0]['message']['content'].strip()
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"[GPT-3.5 Error] {e}"