Update main.py

main.py

@@ -6,10 +6,8 @@ from pydantic import BaseModel
 from datetime import datetime
 import uuid
 import os
-from openai import OpenAI
-
-client = OpenAI()  # Automatically picks up your Hugging Face secret OPENAI_API_KEY
-
+import openai
+ 
 from transformers import pipeline
 import logging, traceback
 from typing import Optional, List, Union
@@ -280,7 +278,7 @@ async def root_cause_analysis(payload: dict, x_api_key: str = Header(None)):
         Suggestion: ...
         """
 
-        response = client.chat.completions.create(
+        response = openai.chat.completions.create(
             model="gpt-4",
             messages=[{"role": "user", "content": prompt}]
         )