update app.py

app.py

@@ -1,12 +1,18 @@
+import os
 import streamlit as st
 import transformers
 import torch
-import os
 import time
 
+# Read Hugging Face API Token from environment variable
+hf_token = os.getenv("HUGGING_FACE_API_TOKEN")
+if not hf_token:
+    st.error("Hugging Face API token not found. Please set the HUGGING_FACE_API_TOKEN environment variable.")
+    st.stop()
+
 # Initialize the model and tokenizer using Hugging Face pipeline
 model_id = "meta-llama/Meta-Llama-3.1-70B"
-pipeline = transformers.pipeline("text-generation", model=model_id)
+pipeline = transformers.pipeline("text-generation", model=model_id, use_auth_token=hf_token)
 
 def generate_response(prompt):
     # Generate a response using the pipeline