Update app.py

app.py

@@ -7,13 +7,18 @@ 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')
-        st.warning("I do sample subsets of CodeXGLUE to finetune !")
-        
+        st.info("Dataset sampled from subsets of CodeXGLUE !")
+
+        self.api_key_area = st.sidebar.text_input(
+            'API key (not necessary for now)',
+            type='password',
+            help="Type in your HuggingFace API key to use this app")
+
         self.API_KEY = os.environ["API_KEY"]
             
         model_parent = st.sidebar.selectbox(
-            label = "Choose Language",
-            options = ["python", "java", "javascript", "php", "ruby"],
+            label = "Choose language",
+            options = ["python", "java", "javascript", "php", "ruby", "go"],
             help="Choose languages",
             )
 
@@ -25,7 +30,7 @@ class LLM_Langchain():
         model_name = return_sum_models(model_parent)
         
         self.checkpoint = st.sidebar.selectbox(
-            label = "Choose Model (nam194/... is my model)",
+            label = "Choose model (nam194/... is my model)",
             options = [model_name, "Salesforce/codet5-base-multi-sum", f"Salesforce/codet5-base-codexglue-sum-{model_parent}"],
             help="Model used to predict",
             disabled=model_name_visibility