Update app.py

app.py

@@ -7,7 +7,8 @@
 @email: [email protected]
 """
 import streamlit as st
-from openai import OpenAI
+#from openai import OpenAI
+from huggingface import transformers
 
 # Initialize session state
 if 'messages' not in st.session_state:
@@ -22,7 +23,8 @@ def create_system_prompt(classification_type, num_to_generate, domain, min_words
     return system_prompt
 
 # OpenAI client setup (replace with your OpenAI API credentials)
-client = OpenAI(api_key='YOUR_API_KEY')
+#client = OpenAI(api_key='YOUR_API_KEY')
+client = Huggingface(api_key='YOUR_API_KEY')
 
 # App title
 st.title("Data Generation for Classification")