Update app.py

app.py

@@ -6,13 +6,10 @@ from models import return_sum_models
 class LLM_Langchain():
     def __init__(self):
         st.warning("Warning: input function needs cleaning and may take long to be processed")
-        st.header('🦜 Code summarization with CodeT5-small + CodeXGLUE dataset')
-        st.warning("Enter your huggingface API key first !")
+        st.header('🦜 Code summarization with CodeT5-small')
+        st.warning("I do sample subsets of CodeXGLUE to finetune !")
         
-        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 = os.environ["API_KEY"]
             
         languages = ["php", "java", "javascript", "python", "ruby"]
         model_parent = st.sidebar.selectbox(