Updated to use token

app.py

@@ -2,6 +2,13 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
+# Retrieve the Hugging Face token from environment variables
+hf_token = os.environ.get("HF_TOKEN")
+
+if not hf_token:
+    st.error("Hugging Face token not found. Please add your HF_TOKEN to the Space secrets.")
+    st.stop()
+
 # Load models and tokenizers
 @st.cache_resource
 def load_model_and_tokenizer(model_name):