Update app.py

app.py

@@ -81,9 +81,9 @@ with st.container():
 with st.container():
   # HuggingFace API KEY input
   API_KEY = st.text_input("Enter your HuggingFace API key", type="password")
-  HF_TOKEN = os.environ.get("HF_TOKEN", None)
+  # HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
-  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+  headers = {"Authorization": f"Bearer {API_KEY}"}
 
   # Selectbox to choose between API URLs
   selected_api_url = st.selectbox("Select Model", options=["bart-large-cnn", "pegasus_indonesian_base-finetune"])