Update model.py

model.py

@@ -7,8 +7,8 @@ def predict_hydrogen_production(params):
     """
     Uses Groq API (Llama 3) to predict hydrogen production based on user inputs.
     """
-    openai.api_key = GROQ_API_KEY
-    
+    client = openai.OpenAI(api_key=GROQ_API_KEY)
+
     prompt = f"""
     Given the electrolysis parameters:
     - Current Density: {params['current_density']} A/cm²
@@ -19,10 +19,11 @@ def predict_hydrogen_production(params):
 
     Predict hydrogen production rate, efficiency, and cost.
     """
-    
-    response = openai.ChatCompletion.create(
+
+    response = client.chat.completions.create(
         model="llama-3-70b",
         messages=[{"role": "user", "content": prompt}]
     )
-    
-    return response["choices"][0]["message"]["content"]
+
+    return response.choices[0].message.content
+