Update app.py

app.py

@@ -3,18 +3,19 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
 
-# Load Hugging Face token from environment variable (Set this in Hugging Face Secrets)
+# Load Hugging Face Token from environment variable
 hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
-if not hf_token:
-    st.error("Hugging Face token not found! Please add it to Secrets in Hugging Face Spaces.")
+
+if hf_token is None:
+    st.error("❌ Hugging Face API Token is missing! Add it in Settings → Secrets.")
     st.stop()
 
-# Load the model and tokenizer
+# Model name
 MODEL_NAME = "google/gemma-2b-it"
+
+# Load tokenizer and model with authentication
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=hf_token)
-model = AutoModelForCausalLM.from_pretrained(
-    MODEL_NAME, torch_dtype=torch.float16, device_map="auto", use_auth_token=hf_token
-)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, device_map="auto", use_auth_token=hf_token)
 
 # Streamlit UI
 st.title("Gemma-2B Code Assistant")