Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,7 +14,7 @@ import streamlit as st
 # It simplifies loading models/tokenizers and running common tasks
 from transformers import pipeline
 
-#hf_token = st.secrets["HF_HUB_TOKEN"]
+hf_token = st.secrets["HF_HUB_TOKEN"]
 #api_key = os.getenv("HF_HUB_TOKEN")
 
 # 1. Cache the pipeline so it loads once
@@ -24,7 +24,7 @@ def get_generator():
     # - "text2text-generation" tells the pipeline we want a seq2seq model (T5 family)
     # - model="google/flan-t5-small" specifies which pretrained model to load
     # The pipeline object wraps both tokenizer and model for you.
-    return pipeline("text2text-generation", model="google/flan-t5-small", use_auth_token=True)
+    return pipeline("text2text-generation", model="google/flan-t5-small", use_auth_token=hf_token)
 
 generator = get_generator()