Update app.py

app.py

@@ -4,7 +4,6 @@ import models
 def main():
     st.set_page_config(layout="wide", page_title="Review Toxicity Checker")
     st.title('Review Toxicity Checker')
-    print(st.secrets["default_hf_api"])
     left_col, right_col = st.columns(2)
 
     content = left_col.empty()
@@ -14,7 +13,7 @@ def main():
             "Select a model option:",
             ["stabilityai/stablelm-zephyr-3b", "mistralai/Mistral-7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.2"]
         )
-        hf_api_key = st.text_input('HF API Key\nhttps://huggingface.co/settings/tokens')
+        #hf_api_key = st.text_input('HF API Key\nhttps://huggingface.co/settings/tokens')
         color = st.color_picker('Highlight Color', '#F44336')
 
     with right_col:
@@ -28,11 +27,11 @@ def main():
         with right_col2:
             clear = st.button('Clear Text')
         if check:
+            hf_api_key = st.secrets["default_hf_api"]
             with st.spinner('Processing review...'):
                 revision = models.revise_review(input_text, hf_api_key, model_id, color)
             try: 
                 with right_col:
-                    print(revision)
                     st.markdown(f""" <div style="border:1px solid white; padding:10px; height:500px; overflow:auto; border-radius:7px;"> {revision['data']['revision']} </div> """, unsafe_allow_html=True)
                     num_revised = int(revision['data']['revised_sentences'])/int(revision['data']['sentence_count'])
                     left_col2, right_col2 = st.columns([1,2])