Update app.py

app.py

@@ -7,13 +7,22 @@ from huggingface_hub import hf_hub_url, HfApi  # Import Hugging Face utilities
 # Title of the Streamlit app
 st.title("Yellow Rust Severity Prediction")
 
-# Load Hugging Face API token from environment
-huggingface_api_token = os.getenv("HUGGINGFACE_TOKEN")
-if huggingface_api_token is None:
-    st.error("YellowRust API token not found in environment. Please set it.")
-    st.stop()
+# Function to get Hugging Face API token securely
+def get_huggingface_api_token():
+    if "HUGGINGFACE_API_TOKEN" not in st.session_state:
+        st.session_state.HUGGINGFACE_API_TOKEN = st.text_input(
+            "Enter your Hugging Face API Token",
+            type="password",
+            label_visibility="collapsed"
+        )
+    return st.session_state.HUGGINGFACE_API_TOKEN
 
 # Set Hugging Face token for authentication
+huggingface_api_token = get_huggingface_api_token()
+if not huggingface_api_token:
+    st.warning("Please enter your Hugging Face API Token.")
+    st.stop()
+
 HfApi().set_access_token(huggingface_api_token)
 
 # Model repository details