Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
-from openai import OpenAI
 import os
 import numpy as np
 import sys
 from dotenv import load_dotenv
 import random
+from huggingface_hub import InferenceClient
 
 # Load environment variables
 load_dotenv()
@@ -19,7 +19,7 @@ def initialize_client():
     if not api_key:
         st.error("HUGGINGFACEHUB_API_TOKEN not found in environment variables.")
         st.stop()
-    return OpenAI(
+    return InferenceClient(
         base_url="https://api-inference.huggingface.co/v1",
         api_key=api_key
     )