Update app.py

app.py

@@ -14,10 +14,10 @@ class LLM_Langchain():
         st.text("Made with Langchain, StreamLit, Hugging Face and 💖")
         st.header('🦜🔗 One stop for Open Source Models')
 
-        self.API_KEY = st.sidebar.text_input(
-            'API Key',
-            type='password',
-            help="Type in your HuggingFace API key to use this app")
+        # self.API_KEY = st.sidebar.text_input(
+        #     'API Key',
+        #     type='password',
+        #     help="Type in your HuggingFace API key to use this app")
 
         self.task_name = st.sidebar.selectbox(
             label = "Choose the task you want to perform",
@@ -73,10 +73,11 @@ class LLM_Langchain():
 
         self.model_kwargs = {
             "temperature": self.temperature,
-            "max_length": self.max_token_length
+            "max_new_tokens": self.max_token_length
         }
 
-        os.environ['HUGGINGFACEHUB_API_TOKEN'] = self.API_KEY
+        # os.environ['HUGGINGFACEHUB_API_TOKEN'] = self.API_KEY
+        os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv("HF_KEY")
 
 
     def generate_response(self, input_text):
@@ -121,8 +122,8 @@ class LLM_Langchain():
     def form_data(self):
         # with st.form('my_form'):
             try:
-                if not self.API_KEY.startswith('hf_'):
-                    st.warning('Please enter your API key!', icon='⚠')
+                # if not self.API_KEY.startswith('hf_'):
+                #     st.warning('Please enter your API key!', icon='⚠')
                 
                 self.selected_option = self.radio_button()
                 self.pdf_uploader()