Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 from PIL import Image
 import requests
+import os
 
 # Load the image classification pipeline
 @st.cache_resource
@@ -13,8 +14,16 @@ pipe_classification = load_image_classification_pipeline()
 # Load the Meta-Llama model and tokenizer for text generation
 @st.cache_resource
 def load_llama_pipeline():
-    tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
-    model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
+    # Retrieve Hugging Face token from environment variables
+    token = os.getenv("HF_AUTH_TOKEN")
+    tokenizer = AutoTokenizer.from_pretrained(
+        "meta-llama/Llama-3.2-3B-Instruct",
+        use_auth_token=token
+    )
+    model = AutoModelForCausalLM.from_pretrained(
+        "meta-llama/Llama-3.2-3B-Instruct",
+        use_auth_token=token
+    )
     return pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 pipe_llama = load_llama_pipeline()