Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,6 +14,8 @@ import streamlit as st
 # It simplifies loading models/tokenizers and running common tasks
 from transformers import pipeline
 
+hf_token = st.secrets["HF_HUB_TOKEN"]
+
 # 1. Cache the pipeline so it loads once
 @st.cache_resource
 def get_generator():
@@ -21,7 +23,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()