Update app.py

app.py

@@ -13,16 +13,12 @@ load_dotenv()
 MAX_TOKENS = 4000
 DEFAULT_TEMPERATURE = 0.5
 
-# Initialize the client
-def initialize_client():
-    api_key = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
-    if not api_key:
-        st.error("HUGGINGFACEHUB_API_TOKEN not found in environment variables.")
-        st.stop()
-    return InferenceClient(
-        base_url="https://api-inference.huggingface.co/v1",
-        api_key=api_key
-    )
+# initialize the client
+client = OpenAI(
+  base_url="https://api-inference.huggingface.co/v1",
+  api_key=os.environ.get('API_KEY')  # Replace with your token
+)
+
 
 # Create supported models
 model_links = {
@@ -49,8 +45,6 @@ st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
 def main():
     st.header('Multi-Models')
 
-    # Initialize client
-    client = initialize_client()
 
     # Sidebar for model selection and temperature
     selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))