Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 from transformers import pipeline
+import os
 import urllib3
 
 # Disable warnings
@@ -10,8 +11,12 @@ urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 session = requests.Session()
 session.verify = False  # Disable SSL verification
 
-# Set up the text generation pipeline
-pipe = pipeline("text-generation", model="meta-llama/Llama-3.1-8B-Instruct", use_auth_token=True, request_session=session)
+# Load the token from an environment variable
+
+
+# Set up the text generation pipeline with the token
+pipe = pipeline("text-generation", model="meta-llama/Llama-3.1-8B-Instruct", 
+                use_auth_token=HUGGINGFACE_TOKEN, request_session=session)
 
 # Streamlit application
 st.title("Text Generation with Hugging Face")