Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,7 +9,7 @@ model_id = "google/gemma-2b-it"
 # 1. READ THE TOKEN: Get the Hugging Face Token from the Space Secrets
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# 2. PASS THE TOKEN to the model and tokenizer loading functions
+# 2. PASS THE TOKEN to the tokenizer loading
 tok = AutoTokenizer.from_pretrained(model_id, token=HF_TOKEN) 
 
 # Simplified Model Loading (trying to avoid complex quantization)
@@ -19,7 +19,7 @@ try:
         model_id, 
         torch_dtype=torch.bfloat16, 
         device_map="auto",
-        token=HF_TOKEN 
+        token=HF_TOKEN # PASS THE TOKEN to the model loading
     )
 except Exception:
     # Fallback to float16 if bfloat16 is not supported